Configurer Docker pour utiliser le conteneur compatible Nvidia GPU [Ubuntu 18.04]
Docker a un support natif pour "Container Acceleration with Nvidia GPUs" depuis la version 19.03.Cette fonctionnalité est rendue disponible à l'aide du NVIDIA Container Toolkit.Ici, effectuez les réglages initiaux nécessaires.
環境
Système d'exploitation: Ubuntu 18.04 LTS
GPU: GTX1080
- 1. Étape XNUMX: Vérifiez les prérequis
- 2. Procédure d'installation de Nvidia docker 2 et confirmation des prérequis [Ubuntu 18.04]
- 3. Étape XNUMX: Installez Docker
- 4. Installez la dernière version de Docker CE sur Ubuntu 18.04 [Document conforme]
- 5. Étape XNUMX: Installez le pilote GPU Nvidia
- 6. Deux façons d'installer le pilote GPU NVIDIA [Ubuntu 18.04]
- 7. Étape XNUMX: rendre le conteneur GPU disponible dans Docker
- 8. Vérifiez le fonctionnement
Étape XNUMX: Vérifiez les prérequis
Avant d'utiliser "l'accélération des conteneurs par GPU" de Docker, vérifions l'environnement.Les conteneurs GPU ne peuvent pas être utilisés dans des environnements trop anciens.
- GNU / Linux x86_64 (version du noyau> 3.10) ← Ubuntu 18.04 donc OK
- GPU NVIDIA avec architecture> Fermi (2.1) ← Si le GPU est destiné au bureau, GTX 600 ou version ultérieure est OK
- Version du pilote NVIDIA ≧ 361.93 ← Voir l'étape XNUMX ci-dessous
Veuillez consulter la page suivante pour plus de détails.
Étape XNUMX: Installez Docker
Si vous n'avez pas encore installé Docker, suivez les étapes ci-dessous pour l'installer.
Étape XNUMX: Installez le pilote GPU Nvidia
Le pilote Nvidia est requis pour exécuter le conteneur GPU (conteneur CUDA).
La version du pilote Nvidia doit être supérieure à la version de CUDA que vous souhaitez utiliser (par exemple, tensorflow nécessite exactement la version CUDA en fonction de la version que vous utilisez).
Tant que vous installez correctement le pilote Nvidia, vous n'avez pas besoin d'installer séparément la boîte à outils CUDA ou la boîte à outils NVIDIA Container.
La méthode d'installation du pilote Nvidia est résumée ci-dessous.
Étape XNUMX: rendre le conteneur GPU disponible dans Docker
Entrez ce qui suit à partir du terminal.
Référence officielle: https://github.com/NVIDIA/nvidia-docker
#Add package repository 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-container-toolkit sudo systemctl restart docker
Vérifiez le fonctionnement
#Testé avec la dernière exécution officielle du docker d'image CUDA --gpus all nvidia / cuda: 9.0-base nvidia-smi
Si tout se passe bien, vous verrez cet affichage.
Pour utiliser le conteneur GPU à partir de maintenant, ajoutez l'option "–gpus" à "docker run".
Si vous venez d'installer docker, il se peut qu'il ne s'exécute que si vous ajoutez sudo avant l'exécution de docker.
[Ubuntu] Paramètres et précautions pour exécuter docker sans la commande sudo
Ce sont les paramètres initiaux requis pour utiliser le conteneur compatible GPU de Docker sur Ubuntu.
Livre de référence