Using Docker

MS-AMP provides a Dockerfile to simplify the process of setting up the development environment. Here are some guides on how to build images and start containers during development

Build image

You need to clone the code first before building the image.

CUDA-12.2

export DOCKER_BUILDKIT=1docker buildx build \  --platform linux/amd64 --cache-to type=inline,mode=max \  --tag msamp-dev-cuda122 --file dockerfile/torch2.1-cuda12.2.dockerfile .

CUDA-11.8

export DOCKER_BUILDKIT=1docker buildx build \  --platform linux/amd64 --cache-to type=inline,mode=max \  --tag msamp-dev-cuda118 --file dockerfile/torch1.14-cuda11.8.dockerfile .

Run container

CUDA-12.2

docker run \  -itd --name=msamp-dev \  --privileged --net=host --ipc=host \  --gpus=all \  -w /root -v /mnt:/mnt \  msamp-dev-cuda122 bash

CUDA-11.8

docker run \  -itd --name=msamp-dev \  --privileged --net=host --ipc=host \  --gpus=all \  -w /root -v /mnt:/mnt \  msamp-dev-cuda118 bash