Cómo instalar CUDA en Ubuntu 20.04

CUDA es una plataforma de cómputo paralelo y un modelo de programación que proporciona una experiencia de usuario extraordinaria cuando se aprovecha la GPU para la informática de uso general cotidiano. CUDA significa C calcular U nificado D dispositivo A arquitectura Es creado por NVIDIA.

CUDA comprende el kit de herramientas de 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 elementos computacionales paralelos. Para investigaciones de aprendizaje profundo y desarrolladores de marcos, use cuDNN para 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 maneras de hacer esto. Puede instalar CUDA desde el repositorio de Ubuntu, que es el método más sencillo, 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. Se recomienda 2 GB de RAM
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 complete la instalación, debemos agregar CUDA a PATH para notificar al shell la ubicación de CUDA. Para ello, especificaremos la RUTA en el archivo .bashrc. Por lo tanto, abra el archivo con 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

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 aquí.

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 marcador a /etc/preferences.d directorio y renómbrelo como cuda-repository-pin-600 .

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

A partir de entonces, descargue el archivo CUDA Debian. Tenga en cuenta que el archivo es bastante grande:2,3 G para ser precisos en el momento de escribir esta guía. Asegúrese de tener una buena conexión a Internet y suficiente 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 el dpkg utilidad 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 de desarrolladores de NVIDIA, que es gratuito.

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.