Чтобы использовать графический процессор в контейнере Docker, нам понадобится nvidia-docker. ( Сик-Хо Цанг @ средний)

Различия между 1.0 и 2.0

Когда я решил установить nvidia-docker 2.0, я обнаружил, что большинство руководств посвящено 1.0, а не очень подробному руководству по 2.0.

Прежде чем говорить об установке nvidia-docker 2.0, я хотел бы сначала упомянуть о разнице между 1.0 и 2.0.

Раньше, когда мы использовали 1.0 , другой демон nvidia-docker использовался для замены docker для запуска образа GPU . И прямо сейчас, с 2.0 нам не нужно заменять docker. Вместо этого мы можем включить nvidia-docker во время выполнения в команде.

Чтобы узнать больше о различиях, вы можете посмотреть
https://github.com/NVIDIA/nvidia-docker/wiki/About-version-2.0

Процедуры установки

Хотя я следовал инструкциям по установке от NVIDIA , у меня были немного другие процедуры, так как я не могу установить nvidia-docker 2.0 в самом начале. Посмотрим, как я его установлю.

0. Если уже есть nvidia-docker 1.0, его нужно удалить. (Но у меня его нет.)

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

      

Если мы попробуем это:

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

Вроде удачно! :)

использованная литература

О nvidia-docker версии 2.0
https://github.com/NVIDIA/nvidia-docker/wiki/About-version-2.0

Установка nvidia-docker (версия 2.0)
https://github.com/nvidia/nvidia-docker/wiki/Installation-(version-2.0)

Использование графического процессора NVIDIA в контейнерах Docker
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/