Add nvidia runtime to docker runtimes
The nvidia
runtime you need, is nvidia-container-runtime
.
Follow the installation instructions here:
https://github.com/NVIDIA/nvidia-container-runtime#installation
Basically, you install it with your package manager first, if it's not present:
sudo apt-get install nvidia-container-runtime
Then you add it to docker runtimes:
https://github.com/nvidia/nvidia-container-runtime#daemon-configuration-file
This option worked for me:
$ sudo tee /etc/docker/daemon.json <<EOF{ "runtimes": { "nvidia": { "path": "/usr/bin/nvidia-container-runtime", "runtimeArgs": [] } }}EOFsudo pkill -SIGHUP dockerd
Check that it's added:
$ docker info|grep -i runtime Runtimes: nvidia runc Default Runtime: runc