Dockerfiles with rolling-release Lambda Stack, designed for use with nvidia-container-toolkit
-
Ensure that you have a docker version > 19.03. On Ubuntu 18.04 and 20.04, you can simply run
sudo apt-get install docker.io. On Ubuntu 16.04, a different OS, or if you prefer to use upstream docker, follow Docker's installation instructions -
If using Lambda Stack on your host machine, install nvidia-container-toolkit with
sudo apt-get install nvidia-container-toolkit. Otherwise, follow NVIDIA's installation instructions
Build the image with the appropriate command for the distribution you wish to use.
sudo docker build -t lambda-stack:16.04 -f Dockerfile.xenial .
sudo docker build -t lambda-stack:18.04 -f Dockerfile.bionic .
sudo docker build -t lambda-stack:20.04 -f Dockerfile.focal .
Note that building these docker images requires acceptance of the cuDNN license agreement
Here's a simple PyTorch test to make sure that your GPUs are usable in your docker images
$ sudo docker run --gpus 2 lambda-stack:20.04 python3 -c "import torch; print(torch.cuda.device_count())"
2