Cómo instalar CUDA en Ubuntu 20.04 LTS

En este tutorial, le mostraremos cómo instalar CUDA en Ubuntu 20.04 LTS. Para aquellos de ustedes que no lo sabían, CUDA son las siglas de Compute Unified Device Architecture. 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.

Este artículo asume que tiene al menos un conocimiento básico de Linux, sabe cómo usar el shell y, lo más importante, aloja su sitio en su propio VPS. La instalación es bastante simple y asume que está ejecutando en la cuenta de root, de lo contrario, es posible que deba agregar ‘sudo‘a los comandos para obtener privilegios de root. Te mostraré la instalación paso a paso de CUDA en Ubuntu 20.04 (Focal Fossa). Puede seguir las mismas instrucciones para Ubuntu 18.04, 16.04 y cualquier otra distribución basada en Debian como Linux Mint.

Install CUDA on Ubuntu 20.04 LTS Focal Fossa

Paso 1. Primero, asegúrese de que todos los paquetes de su sistema estén actualizados ejecutando lo siguiente apt comandos en la terminal.

sudo actualización apta
sudo actualización apta

Paso 2. Instalar CUDA en Ubuntu 20.04.

De forma predeterminada, el paquete CUDA no está disponible en el repositorio base de Ubuntu 20.04. Ahora agregamos el pin del repositorio CUDA a su sistema usando el siguiente comando a continuación:

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

A continuación, movemos el pin a la /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

Después de eso, descargue el archivo CUDA Ubuntu usando el siguiente comando:

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

Luego, usa el dpkg comando para ejecutar el archivo:

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

Finalmente, el paquete enumera e instala el paquete de software CUDA:

sudo actualización apta
sudo apto instalar cuda

Paso 3. Instalación de CuDNN.

Ahora descargamos CuDNN instalado desde la página oficial. Después de registrarse en el Programa para desarrolladores de NVIDIA, puede descargar la última versión de cuDNN:

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

A continuación, copie los siguientes archivos en el 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 *

¡Felicidades! Ha instalado correctamente CUDA. Gracias por usar este tutorial para instalar CUDA NVIDIA en el sistema Ubuntu 20.04 LTS Focal Fossa. Para obtener ayuda adicional o información útil, le recomendamos que consulte el sitio web oficial de NVIDIA.