1、检查显卡驱动
nvidia-smi
若能成功显示则证明显卡驱动安装成功
若没有安装驱动则参考以下链接
可能出现的问题!!!
在安装驱动并重启后,可能电脑出现黑屏(一条小横杠闪烁),这是因为系统没有可用显卡进行图像输出,解决方法如下链接
Ubuntu18-22.04安装和干净卸载nvidia显卡驱动——超详细、最简单_ubuntu安装nvidia显卡驱动_道阻且长行则将至!的博客-CSDN博客
2、安装CUDA
登陆官网CUDA Toolkit Archive | NVIDIA Developer,选择CUDA Toolkit 11.7.0
根据上述的指令安装最快!
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
sudo sh cuda_11.7.0_515.43.04_linux.run
在安装时无需安装driver!
当出现如下界面,则安装过程结束
然后配置环境
sudo gedit ~/.bashrc
source ~/.bashrc
加入以下3条语句
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
注意:CUDA11.7里面没有Samples,因此网上对历史版本验证的方法不可用!只需简单验证即可
nvcc -V
若出现如下界面则表示安装成功
3、安装cudnn
登陆官网选择对应的版本cuDNN Archive | NVIDIA Developer
官方安装教程Installation Guide - NVIDIA Docs
下载完后是一个Deb类型的文件,需要对其进行解压
sudo dpkg -i cudnn-local-repo-ubuntu2004-8.6.0.163_1.0-1_amd64.deb
解压完后会有提示,根据提示把CUDA GPG Key导入
根据指令写
sudo cp /var/cudnn-local-repo-ubuntu2004-8.9.2.26/cudnn-local-6D0A7AE1-keyring.gpg /usr/share/keyrings/
注意!!!网上大部分包括官网教程都会让你下载3个包,但其实已经out了,最新的版本中解压过后,这3个包已经在/var中了,进入cudnn包中可以看到这3个Deb包,只需要依次进行解压即可
验证cudnn是否安装成功
最简单的方法:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
但是这里不推荐,因为新版本的结构貌似已经变化,只需按照如下步骤即可
cp -r /usr/src/cudnn_samples_v8/ $HOME
cd $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN
如出现以下结果则cudnn完整安装
到此,全部安装完成!
参考文章
Ubuntu20.04安装cuDNN(包括WSL)_ubuntu安装cudnn_lconghe的博客-CSDN博客
ubuntu 20.04 安装 高版本cuda 11.7 和 cudnn最新版_cuda11.7对应的cudnn_Keyine?的博客-CSDN博客
Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程(图文)_ubuntu cudnn安装_嵌入式技术的博客-CSDN博客