The list of prerequisites for running nvidia-docker is described below.
For information on how to install Docker for your Linux distribution, please refer to the Docker documentation.
- GNU/Linux x86_64 with kernel version > 3.10
- Docker >= 1.9 (official
docker-engine,docker-ceordocker-eeonly) - NVIDIA GPU with Architecture > Fermi (2.1)
- NVIDIA drivers >= 340.29 with binary
nvidia-modprobe
Your driver version might limit your CUDA capabilities (see CUDA requirements)
- Install the repository for your distribution by following the instructions here.
- Install the
nvidia-dockerpackage
sudo apt-get install nvidia-docker
- Install the repository for your distribution by following the instructions here.
- Install the
nvidia-dockerpackage
sudo yum install nvidia-docker
The package installation will automatically setup the nvidia-docker-plugin and depending on your distribution, register it with the init system.
nvidia-docker run --rm nvidia/cuda nvidia-smi