ubuntu中显卡驱动，cuda，cudnn安装

全能搬运大师2023-12-11 3:04

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