Instalación del controlador de gráficos GPU en Ubuntu20.04

1. Instalar usando la línea de comando

sudo apt-get install nvidia-driver-515

Aviso:

(1) Algunas tarjetas gráficas GPU solo admiten el controlador 455. Si falla la instalación del 515, puede intentar instalar el controlador de gráficos 455.

(2) Una vez completada la instalación, reinicie el entorno: sudo reboot

2. Verifique si la instalación fue exitosa

Entrada de línea de comando: nvidia-smi, el mensaje es el siguiente:

 

3. Instalar CUDA

Dirección de descarga: Descargas de CUDA Toolkit 11.0 Update 3

Seleccione Linux -> x86_64 -> Ubuntu -> 20.04 -> runfile[local]

Comando de instalación:

wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run

Espere a que se complete la descarga:

Una vez completada la descarga de CUDA, instálela:

sudo sh cuda_11.0.3_450.51.06_linux.run

Presione los botones arriba y abajo en el teclado y seleccione continuar:

Ingrese: aceptar cuando el mouse esté sobre el cursor y aparecerá la siguiente página:

Nota: Dado que ya hemos instalado el controlador NVIDIA anteriormente, debemos eliminar el controlador aquí sin instalarlo. Utilice los botones arriba y abajo del teclado para seleccionar el controlador y haga clic en el botón Intro para eliminar la selección.

 

 Como se muestra en la imagen, use los botones arriba y abajo para seleccionar instalar y haga clic en el botón Enter para iniciar la instalación.

Después de una instalación exitosa:

4. Agregue variables de entorno:

Abra el archivo de configuración: sudo vim ~/.bashrc y agregue la siguiente ruta al final del archivo de configuración:

export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=${LD_LIBARY_PATH}:${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}

 Guarde y cierre, luego use el comando: source ~/.bashrc para que surta efecto.

Para verificar si CUDA se instaló correctamente, ingrese el comando: nvcc -V:

total

Resumen: CUDA se instaló correctamente.