Guide d'installation de nvidia-docker2 pour l'utilisation de GPU avec docker

Aperçu

--nvidia-docker2 doit être installé pour utiliser le GPU avec docker. --Nvidia-docker2 a été un peu difficile à installer sur Linux, alors faites une note pour vous-même.

Comment installer nvidia-docker2

guide d'installation


#Inscription au référentiel
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update

# nvidia-Installation de docker2
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

#Test de fonctionnement
docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi

Maintenant, si vous ajoutez l'option d'exécution lors du démarrage du docker et écrivez --runtime = nvidia, le GPU sera reconnu dans le docker.

Recommended Posts

Guide d'installation de nvidia-docker2 pour l'utilisation de GPU avec docker
Implémenter le GPU PyTorch + avec Docker
Cocos2d-x ~ Guide d'installation pour Mac ~
Japaneseize Matplotlib avec Alpine en utilisant Docker
Conseils pour exécuter Go avec Docker
Il est maintenant temps d'installer DB avec Docker! Installation de la base de données pour les débutants sur Docker
Derrière le flyer: utiliser Docker avec Python
Installer Docker
Conseils pour utiliser python + caffe avec TSUBAME
Guide d'installation des outils Python pour Visual Studio
Créer une couche pour AWS Lambda Python dans Docker