☰

Nvidia Docker für Ubuntu 21.04?

tag-icon • tag-icon nvidia docker compatibility 21.04 unsupported-packages
Nvidia Docker für Ubuntu 21.04?

konnte jemand NVidia Docker auf Ubuntu 21.04 installieren? Aufgrund von Kompatibilitätsproblemen kann ich es nicht herunterladen. Anscheinend gibt es noch keine Unterstützung.

Installationsanleitung

Irgendeine Lösung? Vielen Dank

Antwort1

Es ist nicht offiziell, aber Sie können die distributionVariable auf derAnleitungsseitein ubuntu20.04, wie folgt:

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

Der Rest ist gleich:

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

Anschließend können Sie Ihre Installation überprüfen:

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

Sollte ungefähr Folgendes zurückgeben:

+-----------------------------------------------------------------------------+
| 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: Ich muss nur nvidia-dockeretwas Deep Learning mit Tensorflow durchführen, und die Lösung, die ich oben angegeben habe, weist keine Probleme beim Training und bei der Inferenz auf.

Antwort2

Repos hinzufügen: /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 /

Und

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

verwandte Informationen