Untuk menggunakan GPU di dalam container Docker, kita membutuhkan nvidia-docker. ( Sik-Ho Tsang @ Medium)

Perbedaan antara 1.0 dan 2.0

Ketika saya memutuskan untuk menginstal nvidia-docker 2.0, saya baru saja menemukan sebagian besar tutorial berbicara tentang 1.0, dan tidak banyak tutorial rinci untuk 2.0.

Sebelum berbicara tentang instalasi nvidia-docker 2.0, saya ingin menyebutkan perbedaan antara 1.0 dan 2.0 terlebih dahulu.

Di masa lalu, ketika kami menggunakan 1.0 , daemon nvidia-docker lain digunakan untuk menggantikan pekerja galangan untuk menjalankan image GPU . Dan saat ini, dengan 2.0 , kita tidak perlu mengganti buruh pelabuhan. Sebagai gantinya, kita dapat memilih untuk mengaktifkan nvidia-buruh pelabuhan saat runtime dalam perintah.

Untuk mengetahui lebih detail tentang perbedaannya, Anda dapat melihatnya
https://github.com/NVIDIA/nvidia-docker/wiki/About-version-2.0

Prosedur Instalasi

Meskipun saya mengikuti panduan instalasi dari NVIDIA , saya mendapat prosedur yang sedikit berbeda karena saya tidak dapat menginstal nvidia-docker 2.0 di awal. Mari kita lihat bagaimana saya menginstalnya.

0. Jika sudah ada nvidia-docker 1.0, itu perlu dihapus. (Tapi saya tidak memilikinya.)

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

      

Jika kami mencoba ini:

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

Sepertinya berhasil! :)

Referensi

Tentang nvidia-docker Versi 2.0
https://github.com/NVIDIA/nvidia-docker/wiki/About-version-2.0

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

Menggunakan GPU NVIDIA dalam 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/