☰

¿Nvidia Docker para Ubuntu 21.04?

tag-icon • tag-icon nvidia docker compatibility 21.04 unsupported-packages
¿Nvidia Docker para Ubuntu 21.04?

¿Alguien pudo instalar NVidia Docker en Ubuntu 21.04? No me permite descargarlo por problemas de compatibilidad. aparentemente no hay soporte todavía.

Guía de instalación

¿alguna solución? Muchas gracias

Respuesta1

No es oficial, pero puedes cambiar la distributionvariable en elpágina de instruccionesen ubuntu20.04, así:

distribution='ubuntu20.04' \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

El resto es igual:

sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker

Luego, puedes verificar tu instalación:

sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi

Debería devolver algo como esto:

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.51.06    Driver Version: 450.51.06    CUDA Version: 11.0     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  Tesla T4            On   | 00000000:00:1E.0 Off |                    0 |
| N/A   34C    P8     9W /  70W |      0MiB / 15109MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

NB: solo necesito usarlo nvidia-dockerpara hacer un aprendizaje profundo con tensorflow, y la solución que proporcioné anteriormente no tiene problemas para el entrenamiento y la inferencia.

Respuesta2

Agregar repositorios: /etc/apt/sources.list.d/nvidia-docker.list

deb https://nvidia.github.io/libnvidia-container/ubuntu18.04/amd64 /
deb https://nvidia.github.io/nvidia-container-runtime/ubuntu18.04/amd64 /
deb https://nvidia.github.io/nvidia-docker/ubuntu18.04/amd64 /

y

sudo apt-get update
sudo apt-get install -y nvidia-docker2

información relacionada