Instale NVIDIA Docker 2 en Ubuntu 18.04 [verificar disponible 20210225]

 

 

 

Antecedentes: para instalar NVIDIA Transfer, depende del entorno. Para admitir GPU en la ventana acoplable, NVidia solía obtener una nvidia-docker2, ahora se ha actualizado a NVIDIA Container Toolkit.

Consulte github

---

instalación

1. Portal de instalación de Docker

Después de completar la instalación de Docker anterior, comience a configurar la fuente nvidia-docker

1. Agregue una fuente

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
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

•  

3. Instale nvidia-container-toolkit y reinicie la ventana acoplable.

sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

•  

prueba

1. Utilice el mapa oficial de CUDA más reciente para probar nvidia-smi

$ sudo docker run --gpus all nvidia/cuda:10.0-base nvidia-smi

2. Inicie un contenedor que admita GPU duales

$ sudo docker run --gpus 2 nvidia/cuda:10.0-base nvidia-smi

1. Especifique GPU 1, ejecute el contenedor

$ sudo docker run --gpus 2 nvidia/cuda:9.0-base nvidia-smi

1. Especificar el rendimiento del contenedor personal (rara vez se usa)

$ sudo docker run --gpus all,capabilities=utility nvidia/cuda:10.0-base nvidia-smi

Referencia: https://blog.csdn.net/weixin_38369492/article/details/105809571