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 significa 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 elementos computacionales paralelos. Para investigaciones de aprendizaje profundo y desarrolladores de marcos, use cuDNN para aceleración de GPU de alto rendimiento .

Este artículo asume que tiene al menos conocimientos básicos de Linux, sabe cómo usar el shell y, lo que es más importante, aloja su sitio en su propio VPS. La instalación es bastante simple y asume que se están ejecutando en la cuenta raíz, si no, es posible que deba agregar 'sudo ' a los comandos para obtener privilegios de root. Le 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.

Requisitos previos

• Un servidor que ejecuta uno de los siguientes sistemas operativos:Ubuntu 20.04, 18.04 y cualquier otra distribución basada en Debian como Linux Mint.
• Se recomienda que utilice una instalación de sistema operativo nueva para evitar posibles problemas.
• Acceso SSH al servidor (o simplemente abra Terminal si está en una computadora de escritorio).
• Un non-root sudo user o acceder al root user . Recomendamos actuar como un non-root sudo user , sin embargo, puede dañar su sistema si no tiene cuidado al actuar como root.

Instalar CUDA en Ubuntu 20.04 LTS Focal Fossa

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

sudo apt update
sudo apt upgrade

Paso 2. Instalación de CUDA en Ubuntu 20.04.

De manera 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

Luego, movemos 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

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, use 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 apt update
sudo apt install cuda

Paso 3. Instalación de CuDNN.

Ahora descargamos CuDNN instalado desde la página oficial. Después de registrarse en NVIDIA Developer Program, puede descargar la última versión de 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*

¡Felicitaciones! Ha instalado CUDA con éxito. 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 sitio web.