Docker nvidia

De wiki jackbot
Aller à la navigation Aller à la recherche

Pour faire fonctionner une carte nvidia dans un conteneur docker plusieurs points sont nécessaires :

  • installer les pilotes à jour
  • installer cuda
  • installer le paquet : nvidia-container-runtime


Mise à jour des pilotes

Afin d'utiliser les pilotes propriétaires nvidia il est possible d'utiliser le dépôt de nvidia :

sudo add-apt-repository ppa:graphics-drivers/ppa 
sudo apt update 
ubuntu-drivers devices  
sudo apt install nvidia-(numéro du pilote)

Source

Installer Cuda

La procédure est plus ou moins la même que pour la mise à jour des pilotes

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda

Source

Dans le cas où un message d'erreur : "Key is stored in legacy trusted.gpg keyring", suivre la procédure ici

Installer 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

Tester

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

Docker-compose

Avec docker-compose il faut suivre les instructions de cette page