Per utilizzare la GPU all'interno del contenitore Docker, abbiamo bisogno di nvidia-docker. ( Sik-Ho Tsang @ Medium)

Differenze tra 1.0 e 2.0

Quando ho deciso di installare nvidia-docker 2.0, ho appena scoperto che la maggior parte dei tutorial parla di 1.0 e non molto dettagliata per 2.0.

Prima di parlare dell'installazione di nvidia-docker 2.0, vorrei prima menzionare la differenza tra 1.0 e 2.0.

In passato, quando stavamo utilizzando 1.0 , un altro demone nvidia-docker veniva utilizzato per sostituire docker per eseguire l'immagine GPU . E in questo momento, con 2.0 , non è necessario sostituire docker. Invece, possiamo scegliere di abilitare nvidia-docker in fase di esecuzione nel comando.

Per conoscere maggiori dettagli sulle differenze, puoi dare un'occhiata a
https://github.com/NVIDIA/nvidia-docker/wiki/About-version-2.0

Procedure di installazione

Sebbene abbia seguito le linee guida per l' installazione di NVIDIA , ho ottenuto procedure leggermente diverse poiché non posso installare nvidia-docker 2.0 all'inizio. Vediamo come lo installo.

0. Se è già presente nvidia-docker 1.0, è necessario rimuoverlo. (Ma non ce l'ho.)

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

      

Se proviamo questo:

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

Sembra un successo! :)

Riferimenti

Informazioni su nvidia-docker versione 2.0
https://github.com/NVIDIA/nvidia-docker/wiki/About-version-2.0

Installazione di nvidia-docker (versione 2.0)
https://github.com/nvidia/nvidia-docker/wiki/Installation-(version-2.0)

Utilizzo della GPU NVIDIA all'interno dei Docker Containers
https://marmelab.com/blog/2018/03/21/using-nvidia-gpu-within-docker-container.html

安裝 NVIDIA Docker 2 來 讓 容器 使用 GPU
https://kairen.github.io/2018/02/17/container/docker-nvidia-install/