Saltar para o conteúdo principal
Versão: 1.0.0

Instalar CUDA no Linux

Se você estiver usando uma instância de VM habilitada para Nvidia de uma nuvem pública, provavelmente deverá usar a imagem de VM fornecida pela nuvem. Normalmente, ela tem as versões corretas do driver Nvidia e do kit de ferramentas CUDA já instaladas. Continue lendo se precisar instalar o driver Nvidia e o kit de ferramentas CUDA em seu próprio computador.

Ubuntu 22.04​

1 Instale o controlador Nvidia.​

Reconstruir a configuração do 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

Descarregar e instalar o 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

Confirmar que o controlador foi instalado com êxito

nvidia-smi -q | head

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

Registo de data e hora : Sex Oct 27 21:54:05 2023
Versão do driver : 535.104.05
Versão CUDA : 12.2

GPUs anexadas : 1
GPU 00000000:00:1E.0
    Nome do produto : NVIDIA A10G

Desativar o GSP e reiniciar.

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 o kit de ferramentas 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 se o CUDA está instalado.

/usr/local/cuda/bin/nvcc --versão

nvcc: driver do compilador NVIDIA (R) Cuda
Direitos autorais (c) 2005-2023 NVIDIA Corporation
Criado em Ter_Ago_15_22:02:13_PDT_2023
Ferramentas de compilação Cuda, versão 12.2, V12.2.140
Build cuda_12.2.r12.2/compiler.33191640_0

Depois disso, use os dois comandos a seguir para configurar o caminho do ambiente. Você provavelmente deve adicionar essas duas linhas ao seu ~/.bashrc e ~/.bash_profile (ou ~/.zshrc e ~/.perfil) para que os novos terminais e futuros logins ainda possam encontrar esses arquivos de biblioteca CUDA.

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

Mais recursos​

Aqui estão mais scripts que o podem ajudar no caso de ter problemas.

Boa sorte!