参考另一篇文章:Ubuntu20.04安装Nvidia显卡驱动教程_ytusdc的博客-CSDN博客
英伟达官网(最新版):CUDA Toolkit 12.2 Update 1 Downloads | NVIDIA Developer
CUDA历史版本下载地址:CUDA Toolkit Archive | NVIDIA Developer
这里是nvidia给出的官方安装指南(遇到问题时可以查阅):
NVIDIA CUDA Installation Guide for Linux
gcc,如果报错直接安装gcc和g++就好。
sudo apt-get install gcc
sudo apt-get install g++
然后就直接install。等待片刻后,会安装完成,然后在.bashrc里配置环境变量。
sudo gedit ~/.bashrc
在打开文件的最后一行加上:
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
ok,cuda配置结束,正常的测试是用sample里的脚本来测试的,可能是最新版的cuda,没有sample,就假设它装好了。使用nvcc -V测试以下有没有cuda。
然后
source ~/.bashrc
下载cudnn相应版本.
最新版下载地址:Log in | NVIDIA Developer
cudnn历史版本: cuDNN Archive | NVIDIA Developer
官方安装指南: Installation Guide - NVIDIA Docs
下载 CUDA 对应的cudnn 版本
下载完后是一个Deb类型的文件,需要对其进行解压
sudo dpkg -i cudnn*.deb
根据指令写
sudo cp /var/cudnn-local-repo-ubuntu2004-8.5.0.96/cudnn-local-0579404E-keyring.gpg /usr/share/keyrings/
注意!!!网上大部分包括官网教程都会让你下载3个包,但其实已经out了,最新的版本中解压过后,这3个包已经在 /var 中了,进入cudnn包中可以看到这3个Deb包,只需要依次进行解压即可
sudo apt-get update
sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y
sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y
sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y
上面的 X.Y
and 8.x.x.x 换成 /var/cudnn-local-repo-ubuntu2004-8.5.0.96 中文件中的相应版本号
因为新版本的结构貌似已经变化,只需按照如下步骤即可
cp -r /usr/src/cudnn_samples_v8/ $HOME
cd $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN
如出现以下结果则cudnn完整安装
测试时候遇到的错误:
如上图,是因为确实相应的库,安装相应库,再重新测试即可
sudo apt-get install libfreeimage3 libfreeimage-dev