instalação ubuntu20.04-cuda11.0-cudnn8.0.5
Desinstalar driver gráfico
sudo nvidia-uninstall
Confira
sudo find -iname nvidia
Excluir arquivos relacionados
sudo apt-get purge nvidia-*
sudo apt-get remove --purge nvidia-\*
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
Instalar driver gráfico
sudo apt-get install nvidia-390
Modifique nvidia-390 conforme necessário , após a instalação da placa gráfica, reinicie a máquina.
CUDA
sudo dpkg -i cuda-repo-ubuntu1710-9-2-local_9.2.148-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
sudo chmod +x cuda_11.0.3_450.51.06_linux.run
sudo ./cuda_11.0.3_450.51.06_linux.run -no-x-check -no-nouveau-check -no-opengl-files
O driver precisa ser removido, caso contrário, um erro será relatado porque o driver já foi instalado
Você pode ver o site oficial:
download CUDA
/etc/profile adicione o seguinte
sudo vim /etc/profile
ou
sudo vim /etc/bash.bashrc
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=$CUDA_HOME/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
Teste se o cuda foi instalado com sucesso
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
Desinstalar
cd /usr/local/cuda-11.0/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-11.0
Outros
Exibir informações da placa gráfica
lspci|grep NVIDIA
CUDNN
Baixar CUDNN
baixar
instalação de configuração
tar -zxf cudnn-11.0-linux-x64-v8.0.5.39.tgz
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
versão cuda e cudnn
Veja o site oficial
cuda-toolkit
Verifique a versão cuda
cat /usr/local/cuda/version.txt
Ver versão cudnn
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
Observações
Se um erro for relatado, acredite em mim, reinstale o sistema diretamente ou economize mais tempo.