Cómo usar la GPU NVIDIA en contenedores Docker en Ubuntu Top 10.Top 10 LTS

Puede aprobar su GPU NVIDIA en contenedores Docker y ejecutar programas CUDA en su GPU NVIDIA desde estos contenedores Docker. Esta es una característica muy útil para aprender IA (inteligencia artificial). Poder ejecutar códigos de IA (yo.mi. TensorFlow) en los contenedores Docker le ahorrará mucho tiempo. También puede probar sus códigos de IA en diferentes versiones CUDA sin necesidad de cambiar la versión CUDA en su computadora.

Foliar con los conductores de Nvidia y las versiones CUDA en su computadora puede causar problemas como los conductores de Nvidia que no funcionan o que lo dejan con una pantalla de muerte negra/azul. Por lo tanto, siempre es una buena idea dejar su sistema tal como está y hacer cambios en un entorno aislado como Docker Containers. Esto mantiene limpia a su computadora principal (de herramientas de desarrollo innecesarias).

En este artículo, te mostraré cómo configurar Docker CE y Nvidia Docker en Ubuntu 22.04 LTS para que pueda acceder a la GPU NVIDIA de su computadora desde Docker Containers y ejecutar programas CUDA en su GPU NVIDIA.

Tabla de contenido:

1. Verificar si los controladores oficiales de GPU de NVIDIA están instalados en Ubuntu 22.04 LTS
2. Instalación de dependencias de Docker CE en Ubuntu 22.04 LTS
3. Instalación de las teclas GPG de Docker CE en Ubuntu 22.04 LTS
4. Instalación del repositorio de Docker CE en Ubuntu 22.04 LTS
5. Instalación de Docker CE en Ubuntu 22.04 LTS
6. Agregar el Ubuntu 22.04 LTS Iniciar sesión Usuario en el grupo Docker
7. Verificar si Docker CE se instaló correctamente en Ubuntu 22.04 LTS
8. Instalación de las teclas GPG de kit de herramientas de contenedores nvidia en Ubuntu 22.04 LTS
9. Instalación del repositorio de kit de herramientas de contenedores nvidia en Ubuntu 22.04 LTS
10. Instalación de controladores Nvidia-Docker en Ubuntu 22.04 LTS
11. Verificar si se puede acceder a GPU de NVIDIA desde Docker Containers en Ubuntu 22.04 LTS
12. Conclusión
13. Referencias

Verificar si los controladores oficiales de GPU de NVIDIA están instalados en Ubuntu 22.04 LTS:

Antes de comenzar, ejecute el siguiente comando para verificar que tenga una GPU NVIDIA instalada en su computadora.

$ LSPCI | egrep -i 'vga | 3d | pantalla'

En este caso, tengo NVIDIA GTX 1050 TI GPU instalada en mi computadora. Lo más probable es que tenga una GPU NVIDIA diferente instalada en su computadora.

Además, asegúrese de tener los controladores oficiales de Nvidia instalados en su Ubuntu 22.04 Sistema operativo con el siguiente comando:

$ LSMOD | grep nvidia

Si los controladores oficiales de Nvidia están instalados en su Ubuntu 22.04 Sistema operativo, verá las siguientes salidas.

Además, verifique si los controladores oficiales de GPU NVIDIA están trabajando con el siguiente comando:

$ nvidia-smi

Si los controladores oficiales de GPU están funcionando, verá las siguientes salidas. Como puede ver, tengo el controlador oficial de la GPU NVIDIA versión 525.78.01 instalado en mi ubuntu 22.04 máquina.

Si no tiene los controladores oficiales de GPU NVIDIA instalados en su Ubuntu 22.04 máquina y necesita ayuda con eso, consulte el artículo sobre cómo instalar los controladores de Nvidia en Ubuntu 22.04 LTS.

Instalación de dependencias de Docker CE en Ubuntu 22.04 LTS:

Antes de que pueda instalar Docker CE en Ubuntu 22.04, deberá instalar los paquetes de dependencia de Docker CE requeridos en Ubuntu 22.04.

Primero, actualice el caché del repositorio del paquete APT con el siguiente comando:

$ sudo apt actualización

El caché del repositorio del paquete APT debe actualizarse.

Para instalar los paquetes de dependencia requeridos de Docker CE, ejecute el siguiente comando:

$ sudo apt-get install ca certificates curl gnupg lsb-libe

Para confirmar la instalación, presione Y y luego presiona .

Se deben instalar los paquetes de dependencia de Docker CE requeridos.

Instalación de las teclas GPG de Docker CE en Ubuntu 22.04 LTS:

En esta sección, le mostraré cómo instalar la clave GPG del repositorio de paquetes CE Docker en Ubuntu 22.04.

Primero, crea una nueva carpeta /etc/apt/keyrings Con el siguiente comando:

$ sudo mkdir -p/etc/apt/keyrings

Para instalar la clave GPG del repositorio de paquetes CE Docker CE en Ubuntu 22.04, ejecute el siguiente comando:

$ curl -fssl https: // descargar.estibador.com/linux/ubuntu/gpg | Sudo GPG - -Dearmor -O/Etc/Apt/Keyrings/Docker.gpg

Instalación del repositorio de Docker CE en Ubuntu 22.04 LTS:

Para instalar el repositorio de paquetes de Docker CE en Ubuntu 22.04, ejecute el siguiente comando:

$ echo "deb [arqu = $ (dpkg --print-architecture) firmado =/etc/apt/keyrings/docker.gpg] https: // descargar.estibador.com/linux/ubuntu $ (lsb_release -cs) estable "| Sudo tee/etc/apt/fuentes.lista.D/Docker.Lista> /dev /null

Actualice la memoria caché del repositorio de paquetes APT para que los cambios entren en vigencia.

$ sudo apt actualización

Instalación de Docker CE en Ubuntu 22.04 LTS:

Para instalar la última versión de Docker CE en Ubuntu 22.04, ejecute el siguiente comando:

$ sudo apt-get install Docker-ce Docker-CE-Cli Containerd.IO Docker-Compose-Plugin

Para confirmar la instalación, presione Y y luego presiona .

Docker CE y los paquetes de dependencia requeridos se están descargando. Tomará un tiempo completar.

Docker CE y los paquetes de dependencia requeridos se están instalando. Tomará un tiempo completar.

Docker CE y los paquetes de dependencia requeridos deben instalarse en este punto.

Agregar el Ubuntu 22.04 LTS Iniciar sesión Usuario al grupo Docker:

Para crear contenedores Docker y administrarlos sin usar sudo o iniciar sesión como usuario root, deberá agregar su usuario de inicio de sesión al estibador grupo.

Para agregar el usuario de inicio de sesión de su Ubuntu 22.04 al estibador grupo, ejecute el siguiente comando:

$ sudo usermod -ag docker $ (whoami)

Para que los cambios entren en vigencia, reinicie su computadora con el siguiente comando:

$ sudo reinicio

Verificar si Docker CE se instaló correctamente en Ubuntu 22.04 LTS:

Una vez que comience su computadora, ejecute el siguiente comando para verificar que pueda acceder a Docker sin privilegios de Superuser.

Versión de $ Docker

Si todo funciona, debería ver las siguientes salidas.

Como puede ver, estoy ejecutando Docker versión 20.10.23 - La última versión de Docker CE en el momento de este escrito.

Instalación de las teclas GPG de kit de herramientas de contenedores nvidia en Ubuntu 22.04:

En esta sección, le mostraré cómo instalar la tecla GPG del repositorio de paquetes de herramientas de contenedores nvidia en Ubuntu 22.04.

Para instalar la tecla GPG del repositorio de paquetes de herramientas NVIDIA Container en Ubuntu 22.04, ejecute el siguiente comando:

$ curl -fssl https: // nvidia.github.io/libnvidia-continer/gpgkey | Sudo GPG--Dearmor -O/USR/Share/Keyrings/Nvidia-Container-Toolkit-Keyring.gpg

Instalación del repositorio de kit de herramientas de contenedores nvidia en Ubuntu 22.04 LTS:

En esta sección, le mostraré cómo instalar el repositorio del paquete de herramientas de contenedores nvidia en Ubuntu 22.04.

Primero, cree un nuevo archivo fuente APT nvidia-continer-toolkit.lista en el /etc/apt/fuentes.lista.d/ Directorio de la siguiente manera:

$ sudo nano/etc/apt/fuentes.lista.d/nvidia-continer-toolkit.lista

En el nvidia-continer-toolkit.lista Archivo, agregue la siguiente línea y presione + X seguido por Y y Para guardar el archivo.

Deb [firmado-by =/usr/share/keyrings/nvidia-continer-toolkit-keyring.gpg] https: // nvidia.github.io/libnvidia-continer/stable/ubuntu22.04 /$ (arco) /

NOTA: Si estás usando Ubuntu 20.04 lts, ​​entonces todo lo que necesita hacer es agregar la siguiente línea en el nvidia-continer-toolkit.lista Archivo en su lugar y todo lo demás debería funcionar sin cambios.

Deb [firmado-by =/usr/share/keyrings/nvidia-continer-toolkit-keyring.gpg] https: // nvidia.github.io/libnvidia-continer/stable/ubuntu20.04 /$ (arco) /

NOTA: Si estás usando Ubuntu 18.04 lts, ​​entonces todo lo que necesita hacer es agregar la siguiente línea en el nvidia-continer-toolkit.lista Archivo en su lugar y todo lo demás debería funcionar sin cambios.

Deb [firmado-by =/usr/share/keyrings/nvidia-continer-toolkit-keyring.gpg] https: // nvidia.github.io/libnvidia-continer/stable/ubuntu18.04 /$ (arco) /

Para que los cambios entren en vigencia, actualice el caché del repositorio de paquetes APT con el siguiente comando:

$ sudo apt actualización

Instalación de controladores Nvidia-Docker en Ubuntu 22.04 LTS:

Para instalar los controladores Nvidia Docker en Ubuntu 22.04, ejecute el siguiente comando:

$ sudo apt install nvidia-docker2

Para confirmar la instalación, presione Y y luego presiona .

Se deben instalar los controladores Nvidia Docker.

Para que los cambios entren en vigencia, reinicie su computadora con el siguiente comando:

$ sudo reinicio

Verificar si se puede acceder a GPU de NVIDIA desde Docker Containers en Ubuntu 22.04 LTS:

En esta sección, le mostraré cómo crear un contenedor Nvidia Cuda Docker y verificar que el contenedor pueda acceder a la GPU NVIDIA desde su computadora.

Para crear un contenedor Nvidia Cuda 12 Docker basado en Ubuntu 20.04 lts y ejecutar el nvidia-smi Comando en él Una vez que se cree para verificar si puede acceder a la GPU NVIDIA desde su computadora, ejecute el siguiente comando:

$ Docker Run - -Rm - -Gpus All Nvidia/CUDA: 12.0.0-Base-Ubuntu20.04 nvidia-smi

Docker está tirando del Nvidia/Cuda: 12.0.0-Base-Ubuntu20.04 Imagen de Docker Hub. Tomará un tiempo completar.

Una vez que se extrae la imagen Nvidia Cuda Docker y se crea un contenedor, el nvidia-smi El comando se ejecutará e imprimirá la salida en la consola como puede ver en la captura de pantalla a continuación.

El contenedor Docker está usando el Controlador de la GPU NVIDIA 525.78.01^[1] y el CUDA versión 12.0^[2]. Si ve salidas similares, el contenedor Docker puede acceder a la GPU NVIDIA de su computadora.

Si necesita usar versiones anteriores de CUDA, verifique la imagen NVIDIA/CUDA en Docker Hub.

Conclusión:

En este artículo, le he mostrado cómo configurar el repositorio de paquetes CE Docker en Ubuntu 22.04. También le he mostrado cómo instalar la última versión de Docker CE en Ubuntu 22.04. Le he mostrado cómo instalar el repositorio de paquetes de herramientas de contenedores nvidia en Ubuntu 22.04 así como cómo instalar el controlador Nvidia Docker en Ubuntu 22.04. Finalmente, le he mostrado cómo acceder a la GPU NVIDIA de su computadora desde un contenedor Docker.

Referencias:

1. https: // documentos.nvidia.com/datacenter/nube-native/contenedor-toolkit/install-guide.html
2. https: // centro.estibador.com/r/nvidia/cuda/etiquetas