Skip to content

lambdal/lambda-stack-dockerfiles

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Lambda Stack Dockerfiles

Dockerfiles with rolling-release Lambda Stack, designed for use with nvidia-container-toolkit

Installing nvidia-container-toolkit

  1. 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

  2. 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

Building images

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

Testing images

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

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published