ubuntu20.04中cuda11.3和cudnn安装

坚果仙人2024-03-11 11:50

文章目录

一、安装驱动
二、安装cuda11.3
- 方式一：通过自带驱动软件安装
- 方式二：手动下载驱动安装
三、安装cudnn

一、安装驱动

二、安装cuda11.3

方式一：通过自带驱动软件安装

方式二：手动下载驱动安装

1、通过 blacklisting 禁用 Nouveau 驱动

（1）打开终端并切换到 root 用户，编辑 /etc/modprobe.d/blacklist.conf 文件，在文件末尾添加以下内容：

bash 复制代码

blacklist nouveau options nouveau modeset=0

保存并关闭该文件。

（2）重新生成 initramfs：

bash 复制代码

sudo update-initramfs -u

（3）重启系统。

2、安装驱动

（1）驱动下载

GTX 1080ti显卡驱动下载地址：https://us.download.nvidia.com/XFree86/Linux-x86_64/550.54.14/NVIDIA-Linux-x86_64-550.54.14.run

（2）安装

bash 复制代码

sudo chmod +x ./NVIDIA-Linux-x86_64-550.54.14.run
sudo ./NVIDIA-Linux-x86_64-550.54.14.run

三、安装cudnn

参考博客：https://blog.csdn.net/weixin_45303602/article/details/129783168

bash 复制代码

$ tar -xvf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz
$ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
$ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h

验证

bash 复制代码

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