1. 在ubuntu中安装显卡驱动
参考:https://blog.csdn.net/m0_37605642/article/details/119651996
2.在ubuntu中安装cuda
参考:https://blog.csdn.net/m0_61431544/article/details/127007300
2.1 安装cuda
cuda官网:
https://developer.nvidia.com/cuda-toolkit-archive
Cuda11.3:
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run
sudo sh cuda_11.3.0_465.19.01_linux.run
安装的时候如果已经安装了显卡驱动需要把diriver去掉
报错:Failed to verify gcc version. See log at /var/log/cuda-installer.log for details.
$ sudo sh cuda_10.2.89_440.33.01_linux.run
Failed to verify gcc version.
See log at /var/log/cuda-installer.log for details.
解决方法:
sudo sh cuda_10.2.89_440.33.01_linux.run --override
2.2 卸载cuda
sudo apt-get --purge remove "cuda*"
#删除CUDA安装期间创建的任何符号链接:
sudo rm /usr/local/cuda
#删除CUDA安装期间创建的任何其他文件或目录:
sudo rm -rf /usr/local/cuda-<version>
3. 在ubuntu中安装cudnn
3.1 安装cudnn
根据你的cuda选择适合的cudnn版本
下载相应的deb包,然后执行下面命令:
sudo dpkg -i sudo dpkg -i cudnn-local-repo-ubuntu2204-8.5.0.96_1.0-1_amd64.deb
会提示导入库钥:(直接将红色框框中命令执行就行)
sudo apt-get update
打开/var/cudnn-local-repo-ubuntu2204.8.90.文件,执行其中三个deb 文件
3.2 测试cudnn
将 cuDNN 示例文件夹(cudnn_samples_v8)复制到可写路径命令如下;
sudo cp -r /usr/src/cudnn_samples_v8/ /home/user/
#转到可写路径;
cd /home/user/cudnn_samples_v8/mnistCUDNN
#编译 mnistCUDNN 示例;
sudo make clean && make
#运行测试程序:
./mnistCUDNN
若cudnn安装正确会出现:
Test passed!
3.4 cudnn卸载:
sudo apt-get --purge remove libcudnn*
#删除cudnn安装期间创建的任何符号链接:
sudo rm /usr/lib/x86_64-linux-gnu/libcudnn*
#删除cudnn安装期间创建的任何其他文件或目录:
sudo rm -rf /usr/include/cudnn.h /usr/include/cudnn_version.h /usr/lib/x86_64-linux-gnu/libcudnn*
#通过以下命令清除APT缓存:
sudo apt-get clean && sudo apt-get autoremove