算法开发：显卡驱动，cuda和cudnn的安装

喜欢天晴

已于 2023-10-08 11:46:44 修改

阅读量302

点赞数

分类专栏：机器学习算法部署落地专栏文章标签： ubuntu 计算机视觉 python

于 2023-01-03 16:04:55 首次发布

本文链接：https://blog.csdn.net/suanfaxiazai/article/details/128532643

版权

机器学习算法部署落地专栏专栏收录该内容

19 篇文章 0 订阅 ¥299.90 ¥399.90

订阅专栏

超级会员免费看

本文详细介绍了在Ubuntu操作系统中如何禁用Nouveau，安装显卡驱动，接着下载并安装CUDA Toolkit，以及获取和配置cuDNN的过程。通过终端命令行操作，确保CUDA和cuDNN的正确安装，为进行计算机视觉和Python编程提供必要的硬件加速支持。

摘要由CSDN通过智能技术生成

一、显卡驱动安装

1 禁用Nouveau

终端执行

sudo vim /etc/modprobe.d/blacklist.conf

在文本最后添加：

blacklist nouveau
options nouveau modeset=0

终端执行

sudo update-initramfs -u

2 安装驱动

终端执行

sudo ubuntu-drivers devices

sudo ubuntu-drivers autoinstall

3 重启电脑

终端执行

nvidia-smi

终端出现下图内容则表示成功

二、安装cuda

1 查找显卡驱动、cuda之间的版本对应关系，如下图

2 下载CUDA Toolkit

地址：https://developer.nvidia.com/cuda-toolkit-archive

查看系统对应的脚本文件，如下图

终端执行：

wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run

sudo sh cuda_11.3.1_465.19.01_linux.run

脚本启动后，根据提示s设置安装路径，将cuda安装在家目录，我的是

/home/ps/liux_cuda/cuda-11.3

3 下载cuDNN

地址：cuDNN Archive | NVIDIA Developer

选择对应的cudnn版本，如下图

解压tar文件，将出现cuda文件夹，我的目录是

/home/ps/liux-cuda/cudnn/cuda

三、配置cuda

cp /home/ps/liux_cuda/cudnn/cuda/include/cudnn.h  /home/ps/liux_cuda/cuda-11.3/include/  
cp /home/ps/liux_cuda/cudnn/cuda/lib64/libcudnn*  /home/ps/liux_cuda/cuda-11.3/lib64
chmod a+r /home/ps/liux_cuda/cuda-11.3/include/cudnn.h  /home/ps/liux_cuda/cuda-11.3/lib64/libcudnn*

2.终端执行

vim ~/.bashrc

在.bahsrc 文件最后面添加

export PATH=/home/ps/liux_cuda/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/home/ps/liux_cuda/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

终端执行