Docker nvidia

Pour faire fonctionner une carte nvidia dans un conteneur docker il est nécessaire d'installer le paquet :

nvidia-container-runtime

Il faut pour cela ajouter le ppa de nvidia :

curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | sudo apt-key add -

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list

sudo apt-get update && sudo apt-get install nvidia-container-runtime

Source

Il se peut cependant qu'un warning du type : "Key is stored in legacy trusted.gpg keyring", fasse son apparition

Pour y remédier il faut exporter la clef dans un fichier du dossier /etc/apt/trusted.gpg.d Plus de détail ici

Une fois ceci fait on peut suivre les instructions de cette page pour tester si cela à bien fonctionné.

• https://wiki.archlinux.org/index.php/Docker#Run_GPU_accelerated_Docker_containers_with_NVIDIA_GPUs
• https://www.jbnet.fr/systeme/docker/docker-configurer-lutilisation-du-gpu-nvidia.html
• https://marmelab.com/blog/2018/03/21/using-nvidia-gpu-within-docker-container.html
• https://github.com/docker/compose/issues/6691