ubuntu20.04-cuda11.0-cudnn8.0.5 安装

ubuntu20.04-cuda11.0-cudnn8.0.5 安装

卸载显卡驱动

sudo nvidia-uninstall

查看一下

sudo find -iname nvidia

删除相关文件

sudo apt-get purge nvidia-*
sudo apt-get remove --purge nvidia-\*
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

安装显卡驱动

sudo apt-get install nvidia-390

根据需要修改 nvidia-390，显卡安装完，重启机器.

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

需要把Driver给去掉，否则会报错，因为已经装了Driver

可查看官网：
CUDA download

/etc/profile 添加以下内容

sudo vim /etc/profile

或者

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}}

测试cuda是否安装成功

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

卸载

cd /usr/local/cuda-11.0/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-11.0

其他
查看显卡信息

lspci|grep NVIDIA

CUDNN

下载
CUDNN download

配置安装

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*

cuda 和cudnn版本

查看官网
cuda-toolkit

查看 cuda 版本

cat /usr/local/cuda/version.txt

查看 cudnn 版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

备注
如果报错，请相信我，直接重新安装系统，或者节省的时间更多。

参考

• NVIDIA CUDA Toolkit 11.0 安装与卸载(Linux/Ubuntu)