Ir al contenido principal
Versión: 2.0.0

Instalar CUDA en Linux

Si está utilizando una instancia de máquina virtual habilitada para Nvidia de una nube pública, probablemente debería utilizar la imagen de máquina virtual proporcionada por la nube. Normalmente ya tiene instaladas las versiones correctas del controlador Nvidia y del kit de herramientas CUDA. Sigue leyendo si necesitas instalar el controlador Nvidia y el kit de herramientas CUDA en tu propia máquina.

Ubuntu 22.04​

1 Instale el controlador Nvidia.​

Reconstruir la configuración de grub:

sudo apt-get install -y gcc make linux-headers-$(uname -r)
cat << EOF | sudo tee --append /etc/modprobe.d/blacklist.conf
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv
EOF
sudo sed -i 's/GRUB_CMDLINE_LINUX=""/GRUB_CMDLINE_LINUX="rdblacklist=nouveau"/' /etc/default/grub
sudo update-grub

Descargar e instalar el controlador Nvidia

wget https://storage.googleapis.com/nvidia-drivers-us-public/GRID/vGPU16.1/NVIDIA-Linux-x86_64-535.104.05-grid.run
sudo sh NVIDIA-Linux-x86_64-535.104.05-grid.run

Confirme que el controlador se ha instalado correctamente

nvidia-smi -q | head

==============NVSMI LOG==============

Fecha y hora : Fri Oct 27 21:54:05 2023
Versión del controlador : 535.104.05
Versión CUDA : 12.2

GPUs conectadas : 1
GPU 00000000:00:1E.0
    Nombre del producto : NVIDIA A10G

Desactive el SPG y reinicie.

sudo touch /etc/modprobe.d/nvidia.conf
echo "options nvidia NVreg_EnableGpuFirmware=0" | sudo tee --append /etc/modprobe.d/nvidia.conf
sudo reboot

2 Instale el kit de herramientas CUDA.​

wget https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run
sudo sh cuda_12.2.2_535.104.05_linux.run --silent --override --toolkit --samples --toolkitpath=/usr/local/cuda-12 --samplespath=/usr/local/cuda --no-opengl-libs

Confirme que CUDA está instalado.

/usr/local/cuda/bin/nvcc --version

nvcc: controlador del compilador de NVIDIA (R) Cuda
Copyright (c) 2005-2023 NVIDIA Corporation
Creado el Tue_Aug_15_22:02:13_PDT_2023
Herramientas de compilación de Cuda, versión 12.2, V12.2.140
Build cuda_12.2.r12.2/compiler.33191640_0

Después de eso, utilice los dos comandos siguientes para configurar la ruta del entorno. Probablemente deberías añadir estas dos líneas a tu ~/.bashrc y ~/.bash_profile (o ~/.zshrc y ~/.profile) para que los nuevos terminales y futuros usuarios puedan encontrar estos archivos de la biblioteca CUDA.

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

Más recursos​

Aquí tienes más guiones que podrían ayudarte en caso de que te quedes atascado.