Это 15-й день моего участия в ноябрьском испытании обновлений. Узнайте подробности события:Вызов последнего обновления 2021 г.
Докер Nvidia - это инструмент докера, который упакован графической картой nvidia на основе докера.Для этого необходимо установить драйвер видеокарты Nvidia и докер на компьютер.После настройки докера Nvidia докер может использовать графический процессор. В этой статье описывается установка и использование докера Nvidia.
окрестности
- Linux 16.04 64-битная операционная система
- Драйвер видеокарты 450.80.02
- CUDA версии 11.0
- докер установить версию 19.03.4
Процесс установки
- добавить репозиторий пакетов
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
- Установите конфигурацию загрузочного докера nvidia-docker2.
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd
- Загрузите официальное тестовое изображение
docker run --runtime=nvidia --rm nvidia/cuda:11.0-base nvidia-smi
В этот момент будет отображаться информация о видеокарте, указывающая, что докер nvidia был успешно создан и правильно выполнен внутри.
nvidia-smiЗаказ.
- Проверьте установку nvidia-docker
$ sudo apt show nvidia-docker2
->
Package: nvidia-docker2
Version: 2.5.0-1
Priority: optional
Section: utils
Maintainer: NVIDIA CORPORATION <cudatools@nvidia.com>
Installed-Size: 27.6 kB
Depends: nvidia-container-runtime (>= 3.4.0), docker-ce (>= 18.06.0~ce~3-0~ubuntu) | docker-ee (>= 18.06.0~ce~3-0~ubuntu) | docker.io (>= 18.06.0)
Breaks: nvidia-docker (<< 2.0.0)
Replaces: nvidia-docker (<< 2.0.0)
Homepage: https://github.com/NVIDIA/nvidia-docker/wiki
Download-Size: 5,840 B
APT-Manual-Installed: yes
APT-Sources: https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64 Packages
Description: nvidia-docker CLI wrapper
Replaces nvidia-docker with a new implementation based on
nvidia-container-runtime
N: There are 50 additional records. Please use the '-a' switch to see them.
Появится аналогичное сообщение, указывающее, что установка прошла успешно
Использование образа докера nvidia
- Образ, созданный с той же версией драйвера, может не найти команду nvidia-smi в докере nvidia под новым драйвером
- Чтобы использовать gpu под новым докером, я обходным путем использую тестовый образ в качестве начального образа.
nvidia-docker run -it --name first_container nvidia/cuda:11.0-base /bin/bash
Таким образом, тестовый образ создается в контейнере, к которому можно обращаться и изменять в любое время, а образ можно сохранить на основе контейнера.
docker commit -m "nvidia docker image init" first_container my_image:1.0