J'essaie d'obtenir un support GPU sur mon conteneur sans le nvidia-docker p>
Je sais avec le Docker Nvidia, je dois juste utiliser --Runtime = nvidia mais mes circonstances actuelles ne permettent pas d'utiliser NVIDIA-Docker P>
J'ai essayé d'installer le pilote Nvidia, Cuda, Cudnn sur mon conteneur mais il échoue. P>
Comment puis-je utiliser GPU TensorFlow sans Nvidia Docker sur mon conteneur? P>
4 Réponses :
Vous pouvez utiliser x11docker
L'exécution d'une image docker sur X avec gpu est aussi simple que
x11docker --gpu imagename
Vous ne pouvez pas simplement installer les pilotes nvidia dans un conteneur Docker. Le conteneur doit avoir accès au matériel. Bien que je ne sois pas certain, mais les montures pourraient vous aider à résoudre ce problème. Voir - https://docs.docker.com/storage/
Vous pouvez utiliser anaconda pour installer et utiliser Tensorflow-gpu.
Assurez-vous que les derniers pilotes nvidia sont installés. Installez Anaconda 2 ou 3 depuis le site officiel. https://www.anaconda.com/distribution/
Créez un nouvel environnement et installez tensorflow-gpu et cudatoolkit.
$ conda create -n tf-gpu tensorflow-gpu python cudnn cudatoolkit
Vous pouvez également spécifier la version de l'application.
Par exemple $ conda create -n tf-gpu tensorflow-gpu python = 3.5 cudnn cudatoolkit = 8
Si vous ne pouvez pas passer --runtime = nvidia
comme option de ligne de commande (par exemple docker-compose
), vous pouvez définir le runtime par défaut dans le Docker fichier de configuration du démon /etc/docker/daemon.json
:
{ "default-runtime": "nvidia" }
Vous serez heureux de savoir que la dernière version de Docker est désormais compatible avec les GPU nvidia. Vous devrez utiliser l'indicateur --device pour spécifier votre pilote Nvidia. Voir - Comment utiliser le GPU dans un conteneur Docker
Auparavant, vous deviez installer nvidia-docker qui était un simple docker avec une fine couche d'abstraction pour les GPU nvidia. Voir - Nvidia Docker < / p>
En raison de la configuration matérielle requise, cela nécessiterait probablement des conteneurs privilégiés.