Um die GPU im Docker-Container zu verwenden, benötigen wir nvidia-docker. ( Sik-Ho Tsang @ Medium)

Unterschiede zwischen 1,0 und 2,0

Als ich mich entschied, nvidia-docker 2.0 zu installieren, habe ich gerade festgestellt, dass die meisten Tutorials über 1.0 sprechen und nicht sehr detaillierte Tutorials für 2.0.

Bevor ich über die Installation von nvidia-docker 2.0 spreche, möchte ich zuerst auf den Unterschied zwischen 1.0 und 2.0 eingehen.

In der Vergangenheit, als wir 1.0 verwendeten , wurde ein anderer nvidia-docker-Daemon verwendet, um Docker zu ersetzen, um das GPU-Image auszuführen . Und im Moment müssen wir mit 2.0 Docker nicht ersetzen. Stattdessen können wir nvidia-docker zur Laufzeit im Befehl aktivieren.

Um mehr Details über die Unterschiede zu erfahren, können Sie einen Blick auf werfen
https://github.com/NVIDIA/nvidia-docker/wiki/About-version-2.0

Installationsverfahren

Obwohl ich die Installationsrichtlinie von NVIDIA befolgt habe, habe ich ein etwas anderes Verfahren erhalten, da ich nvidia-docker 2.0 nicht ganz am Anfang installieren kann. Mal sehen, wie ich es installiere.

0. Wenn nvidia-docker 1.0 bereits vorhanden ist, muss es entfernt werden. (Aber ich habe es nicht.)

docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge nvidia-docker

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

      

sudo apt-get install nvidia-docker2
sudo pkill -SIGHUP dockerd

      

docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

      

apt-cache madison nvidia-docker2 nvidia-container-runtime

      

sudo apt-get install nvidia-docker2=2.0.3+docker18.03.1–1
sudo pkill -SIGHUP dockerd

      


      

docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

      

Wenn wir das versuchen:

sudo docker run --rm --runtime=nvidia nvidia/cuda:9.0-devel nvcc --version

Es scheint erfolgreich! :)

Verweise

Über nvidia-docker Version 2.0
https://github.com/NVIDIA/nvidia-docker/wiki/About-version-2.0

nvidia-docker-Installation (Version 2.0)
https://github.com/nvidia/nvidia-docker/wiki/Installation-(version-2.0)

Verwenden von NVIDIA-GPU in Docker-Containern
https://marmelab.com/blog/2018/03/21/using-nvidia-gpu-within-docker-container.html

Verwenden Sie NVIDIA Docker 2 für die GPU
https://kairen.github.io/2018/02/17/container/docker-nvidia-install/