Cómo usar la GPU NVIDIA en los contenedores Docker en Debian Top 10

Cómo usar la GPU NVIDIA en los contenedores Docker en Debian Top 10
Puede aprobar su GPU NVIDIA en los contenedores Docker y ejecutar los programas CUDA en su GPU NVIDIA desde estos contenedores Docker. Esta es una característica muy útil para aprender IA (inteligencia artificial). Poder ruisear los 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. Siempre es una buena idea dejar su sistema tal como está y hacer cambios en un entorno aislado como los contenedores Docker. Esto mantiene limpia a su computadora principal (de herramientas de desarrollo innecesarias).

En este artículo, le mostraremos cómo instalar Docker CE y Nvidia Docker en Debian 11 para que pueda acceder a la GPU NVIDIA de su computadora desde los contenedores Docker y ejecutar los programas CUDA en su GPU NVIDIA.

Tabla de contenido:

  1. Comprobación de si los conductores oficiales de GPU de NVIDIA están instalados en Debian 11
  2. Instalación de las dependencias de Docker CE en Debian 11
  3. Instalación de las teclas GPG de Docker CE en Debian 11
  4. Instalación del repositorio de Docker CE en Debian 11
  5. Instalación del Docker CE en Debian 11
  6. Agregar el usuario de inicio de sesión de Debian 11 al grupo Docker
  7. Verificar si el Docker CE se instala correctamente en Debian 11
  8. Instalación de las teclas GPG de kit de herramientas de contenedores nvidia en Debian 11
  9. Instalación del repositorio de kit de herramientas de contenedores NVIDIA en Debian 11
  10. Instalación de los controladores Nvidia-Docker en Debian 11
  11. Comprobación de si se puede acceder a la GPU de NVIDIA desde los contenedores Docker en Debian 11
  12. Conclusión
  13. Referencias

Comprobación de si los conductores oficiales de GPU de NVIDIA están instalados en Debian 11

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

$ LSPCI | Grep VGA

En este caso, tenemos la GPU NVIDIA GTX 1050 TI instalada en nuestra 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 sistema operativo Debian 11 con el siguiente comando:

$ LSMOD | grep nvidia

Si los controladores oficiales de NVIDIA están instalados en su sistema operativo Debian 11, 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, tenemos el controlador oficial de la GPU NVIDIA versión 470.161.03 instalado en nuestra máquina Debian 11:

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

Instalación de las dependencias de Docker CE en Debian 11

Antes de que pueda instalar Docker CE en Debian 11, debe instalar los paquetes de dependencia de Docker CE requeridos en Debian 11.

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 presione .

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

Instalación de las teclas GPG de Docker CE en Debian 11

En esta sección, le mostraremos cómo instalar la clave GPG del repositorio de paquetes CE Docker en Debian 11.

Primero, cree una carpeta nueva/etc/apt/keyrings con el siguiente comando:

$ sudo mkdir -p/etc/apt/keyrings

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

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

Instalación del repositorio de Docker CE en Debian 11

Para instalar el repositorio de paquetes CE Docker en Debian 11, ejecute el siguiente comando:

$ echo "deb [arqu = $ (dpkg --print-architecture) firmado =/etc/apt/keyrings/docker.gpg] https: // descargar.estibador.com/linux/Debian $ (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 del Docker CE en Debian 11

Para instalar la última versión del Docker CE en Debian 11, 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 presione .

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

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

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

Agregar el usuario de inicio de sesión de Debian 11 al grupo Docker

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

Para agregar el usuario de inicio de sesión de su Debian 11 al grupo Docker, 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 el Docker CE se instala correctamente en Debian 11

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

Versión de $ Docker

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

Como puede ver, estamos 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 Debian 11

En esta sección, le mostraremos cómo instalar la tecla GPG del repositorio de paquetes de herramientas de contenedores NVIDIA en Debian 11.

Para instalar la clave GPG del repositorio de paquetes de kit de herramientas NVIDIA Container en Debian 11, 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 Debian 11

En esta sección, le mostraremos cómo instalar el repositorio de paquetes de herramientas de contenedores nvidia en Debian 11.

Primero, cree un nuevo archivo fuente APT nvidia-Container-Toolkit.Lista en las fuentes/etc/apt/.lista.D/ Directorio de la siguiente manera:

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

En el nvidia-continer-toolkit.Lista del archivo, agregue la siguiente línea y presione + x seguido de Y y para guardar el archivo:

Deb [firmado-by =/usr/share/keyrings/nvidia-continer-toolkit-keyring.gpg] https: // nvidia.github.IO/Libnvidia-Container/Stable/Debian11/$ (Arch)/

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 los controladores Nvidia-Docker en Debian 11

Para instalar los controladores Nvidia Docker en Debian 11, ejecute el siguiente comando:

$ sudo apt install nvidia-docker2

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

Se deben instalar los controladores Nvidia Docker.

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

$ sudo reinicio

Comprobación de si se puede acceder a la GPU de NVIDIA desde los contenedores Docker en Debian 11

En esta sección, le mostraremos 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 ejecute el comando nvidia-smi 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 de Nvidia/Cuda: 12.0.0-Base-Ubuntu20.04 Imagen del Docker Hub. Tarda un tiempo para completar.

Una vez que se extrae la imagen NVIDIA CUDA Docker y se crea un contenedor, el comando NVIDIA-SMI se ejecuta e imprime la salida en la consola como puede ver en la siguiente captura de pantalla:

El contenedor Docker utiliza el controlador de GPU NVIDIA 525.78.01[1] y la versión cuda 12.0[2]. Si ve algunas salidas similares, el contenedor Docker puede acceder a la GPU NVIDIA de su computadora.

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

Conclusión

Le mostramos cómo instalar el repositorio de paquetes CE Docker en Debian 11. También le mostramos cómo instalar la última versión del Docker CE en Debian 11. Le mostramos cómo instalar el repositorio de paquetes de herramientas Nvidia Container en Debian 11, así como cómo instalar el controlador Nvidia Docker en Debian 11. Finalmente, le mostramos 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