[LINUX] Comment créer un environnement NVIDIA Docker

introduction

Notez comment créer un environnement NVIDIA Docker à partir du 4 novembre 2020. Avant de le savoir, CUDA n'a pas été reconnu, alors j'ai décidé de le remettre en place depuis le conducteur.

USER@HOST:~$nvidia-smi
Wed Nov  4 14:41:54 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.64       Driver Version: 430.64       CUDA Version: N/A      |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 108...  Off  | 00000000:01:00.0  On |                  N/A |
|  0%   35C    P8    12W / 275W |     62MiB / 11175MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1609      G   /usr/lib/xorg/Xorg                            59MiB |
+-----------------------------------------------------------------------------+

Effacer l'ancien environnement

#Désinstaller le CUDA actuel
$ sudo apt purge cuda*
$ sudo apt purge nvidia-cuda-*
$ sudo apt purge libcuda*
#Complètement supprimé
$ sudo apt-get purge nvidia*
$ sudo apt-get autoremove
$ sudo apt-get autoclean
$ sudo rm -rf /usr/local/cuda*

Sujet principal

Vous pouvez installer CUDA, Docker et nvidia-docker2 en suivant les 3 étapes ci-dessous. Jusqu'à présent, le pilote Nvidia était installé séparément, mais si vous installez CUDA, le pilote correspondant sera également installé.

1. Installez le dernier pilote CUDA

Installer conformément au site Web de la boîte à outils CUDA (https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=debnetwork)

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin
$ sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/ /"
$ sudo apt-get update
$ sudo apt-get -y install cuda #(* Lors de la spécification de la version, cuda-10-Ecrire comme 2)

2. Installez Docker

Installez selon la méthode d'installation officielle de Docker

$ curl -fsSL https://get.docker.com -o get-docker.sh
$ sudo sh get-docker.sh

Après l'installation, configurez le service docker pour qu'il démarre et démarre automatiquement

$ sudo systemctl start docker && sudo systemctl enable docker

Si vous souhaitez que la commande docker soit exécutable sans sudo, ajoutez l'utilisateur cible au groupe docker comme suit:

$ sudo usermod -aG docker $USER

3. Installez le package NVIDIA Container Toolkit (nvidia-docker2)

Installez selon Méthode d'installation officielle

$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && 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
$ sudo apt-get install -y nvidia-docker2
$ sudo systemctl restart docker

Après le redémarrage, vérifiez le fonctionnement

$ sudo reboot
$ nvidia-smi

Terminé lorsque le pilote Nvidia et la version CUDA sont affichés.

Référence [Que se passe-t-il avec NVIDIA Docker maintenant? (Version 20.09)](https://medium.com/nvidiajapan/nvidia-docker-%E3%81%A3%E3%81%A6%E4%BB%8A%E3%81%A9%E3%81%86 % E3% 81% AA% E3% 81% A3% E3% 81% A6% E3% 82% 8B% E3% 81% AE-20-09-% E7% 89% 88-558fae883f44)

Recommended Posts

Comment créer un environnement NVIDIA Docker
Comment utiliser Tensorflow dans un environnement Docker
Introduction à docker Création d'un environnement ubuntu dans ubuntu
Comment créer un agent de ressources compatible OCF
[Note] Comment créer un environnement de développement Ruby
[Note] Comment créer un environnement de développement Mac
Comment créer un article à partir de la ligne de commande
Créer un environnement avec virtualenv
Comment créer un environnement Django (python) sur Docker
Comment créer un téléchargeur d'image avec Bottle (Python)
Créez un environnement pour "Deep Learning from scratch" avec Docker
Makefile minimal et buildout.cfg pour créer un environnement avec buildout
Présentation de l'image Docker de Kaggle sur Windows pour créer un environnement
Créer une application Todo avec Django ① Créer un environnement avec Docker
[Python Kivy] Comment créer un fichier exe avec pyinstaller
Comment créer un fichier ISO (image CD) sous Linux
[Django] Mémo pour créer un environnement de Django + MySQL + Vue.js [Python]
Vue d'ensemble de l'environnement virtuel Python et comment le créer
Comment créer un package Conda
Comment créer un pont virtuel
Comment créer / supprimer des liens symboliques
Comment créer un Dockerfile (basique)
Comment supprimer un conteneur Docker
Comment écrire une image de base Docker
Comment créer un fichier de configuration
Comment se connecter à Docker + NGINX
J'ai essayé de créer un environnement de MkDocs sur Amazon Linux
Comment déboguer Dash (Flask) dans Docker + VSCode + environnement de connexion à distance
Comment utiliser Jupyter Notebook sans polluer votre environnement avec Docker
Créez un environnement d'apprentissage automatique arbitraire avec GCP + Docker + Jupyter Lab
Apprenez à utiliser Docker en créant un environnement pour Django + MySQL
Comment créer facilement un environnement où le code Python peut s'exécuter sur Jupyter sans polluer l'environnement local
Comment créer un clone depuis Github
Comment créer un environnement de traduction sphinx
Comment créer un dossier git clone
python3 Comment installer un module externe
Comment créer un fichier * .spec pour pyinstaller.
Comment convertir Python en fichier exe
J'ai créé un environnement Anaconda à l'aide de Docker!
Comment préparer l'environnement de développement Python [Mac]
Créer un environnement Python + uWSGI + Nginx avec Docker
Créer un environnement OpenCV3 + python3 sur OSX
Comment créer un référentiel à partir d'un média
Comment ajouter un module python à l'environnement anaconda
Comment installer python3 avec docker centos
Préparer un environnement pour toucher les fichiers au format grib2 avec python (édition Docker)
[Environnement de développement] Comment créer un ensemble de données proche de la base de données de production
Comment créer des exemples de données CSV avec hypothèse
Comment créer un pilote de périphérique Linux intégré (11)
Comment créer des fichiers volumineux à haute vitesse
Comment partager un environnement virtuel [About requirements.txt]
Pour les débutants à créer un environnement Anaconda. (Note)
Comment créer un environnement lorsque vous souhaitez utiliser python2.7 après l'installation d'Anaconda3
Comment créer un pilote de périphérique Linux intégré (1)
Comment créer un pilote de périphérique Linux intégré (4)