Cómo instalar CUDA en Ubuntu 20.04 | Ubuntu 18.08

Cómo instalar CUDA en Ubuntu 20.04 | Ubuntu 18.08

 

CUDA es una plataforma de computación paralela y un modelo de programación que brinda una experiencia de usuario extraordinaria al aprovechar la GPU para la computación de uso general de todos los días. CUDA son las siglas de C ompute U nified D evice A rchitecture. Es creado por NVIDIA.

CUDA comprende el kit de herramientas CUDA (compilador, perfil y depurador), el controlador de software y el SDK de CUDA. La capa de software brinda acceso directo al conjunto de instrucciones virtuales de la GPU y a los elementos computacionales paralelos. Para las investigaciones de aprendizaje profundo y los desarrolladores de marcos, use cuDNN para la aceleración de GPU de alto rendimiento. Es una biblioteca de primitivas para redes neuronales profundas.

En esta guía, aprendemos cómo instalar CUDA y CuDNN en Ubuntu 20.04 . Hay dos formas de hacerlo. Puede instalar CUDA desde el repositorio de Ubuntu, que es el enfoque más fácil, o instalar desde el repositorio de CUDA, que proporciona la última versión de CUDA.

Requisitos previos:

  1. Mínimo 20 GB de espacio libre en disco
  2. 2 GB de RAM recomendados
  3. Debe tener instalado el controlador de gráficos NVIDIA. Es posible que el controlador que viene con CUDA no sea la última versión.

Método 1: instalar CUDA desde el repositorio de Ubuntu

Comience actualizando las listas de paquetes en su instancia de Ubuntu 20.04.

$ sudo apt update

A continuación, instale el kit de herramientas CUDA utilizando el administrador de paquetes APT de la siguiente manera. Tenga en cuenta que la instalación es intensiva y, como tal, asegúrese de tener una conexión a Internet rápida y estable y un mínimo de 10 G de espacio en disco.

$ sudo apt install nvidia-cuda-toolkit

Una vez que se completa la instalación, debemos agregar CUDA a PATH para notificar al shell la ubicación de CUDA. Para hacer esto, especificaremos la RUTA en el archivo .bashrc. Entonces, abra el archivo usando su editor de texto de línea de comandos.

$ sudo vim ~/.bashrc

Al final del archivo, copie y pegue las siguientes líneas de código.

# set PATH for cuda 10.1 installation

if [ -d "/usr/local/cuda-10.1/bin/" ]; then

export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

fi

A continuación, vuelva a cargar los cambios en el archivo .bashrc.

$ source ~/.bashrc

Ahora verifique y confirme la versión de CUDA instalada usando el siguiente comando.

$ nvcc --version
Cómo instalar CUDA en Ubuntu 20.04

Método 2: instalar CUDA desde el repositorio de CUDA

Como se mencionó anteriormente, el repositorio de CUDA proporciona la última versión de CUDA. Por lo tanto, si desea instalar la última versión, siga las instrucciones que se describen en este documento.

Primero, descargue el pin del repositorio de CUDA:

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

A continuación, mueva el pin al /etc/preferences.ddirectorio y cámbiele el nombre cuda-repository-pin-600.

$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

Cómo instalar CUDA en Ubuntu 20.04

A partir de entonces, descargue el archivo CUDA Debian. Se le notifica que el archivo es bastante grande – 2.3G para ser precisos al momento de escribir esta guía. Asegúrese de tener una buena conexión a Internet y un amplio espacio en disco.

$ wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

Una vez descargado, use la dpkgutilidad para ejecutar el archivo.

sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

A continuación, instale la clave CUDA GPG de la siguiente manera:

$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub

Finalmente, actualice las listas de paquetes e instale CUDA usando el administrador de paquetes APT.

$ sudo apt update
$ sudo apt install cuda -y

Instalar CuDNN

CuDNN no viene con Cuda. Para descargar CuDNN , debe registrarse para convertirse en miembro del Programa para desarrolladores de NVIDIA, que es gratuito.

Cómo instalar CUDA en Ubuntu 20.04

Después de registrarse en el Programa para desarrolladores de NVIDIA, puede descargar la última versión de cuDNN. Aquí estoy descargando la biblioteca cuDNN para Linux.

Primero, extraiga el paquete cuDNN:

$ tar -xzvf cudnn-10.2-linux-x64-v8.2.4.15.tgz

Luego copie los siguientes archivos al directorio CUDA:

$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

Conclusión

En este tutorial, aprendimos cómo instalar CUDA y cuDNN en Ubuntu 20.04. Gracias por leer, deje sus comentarios y sugerencias.

Deja un comentario

Tu dirección de correo electrónico no será publicada.

Este sitio usa Akismet para reducir el spam. Aprende cómo se procesan los datos de tus comentarios.