
These are the installation instructions for Ubuntu:
- Follow the docker-ce installation from (removing docker.io):
https://docs.docker.com/install/linux/docker-ce/ubuntu - Install nvidia-docker:
https://github.com/NVIDIA/nvidia-docker
Test the installation running
$ docker run --runtime=nvidia --rm nvidia/cuda:latest nvidia-smior the equivalent nvidia-docker wrapper call
$ nvidia-docker run --rm nvidia/cuda:latest nvidia-smi
The wrapper allows to easily select which of the host's GPUs are visible from the docker instance, in this example only the first two GPUs
$ NV_GPU=0,1 nvidia-docker run --rm -v /home/foo/data:/data -it nvidia/cuda:latest bash
No comments:
Post a Comment