ubuntu18.04系统安装cuda、cudnn

头不秃还能学

已于 2022-04-22 16:43:12 修改

阅读量2.9k

点赞数 3

分类专栏： ros机械臂开发文章标签： ubuntu

于 2022-04-16 16:24:41 首次发布

本文链接：https://blog.csdn.net/qq_43124746/article/details/124214360

版权

ros机械臂开发专栏收录该内容

3 篇文章

订阅专栏

1.安装Nvidia显卡驱动

1.1添加源并更新

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

1.2查询网卡驱动

ubuntu-drivers devices

在这里插入图片描述 1.3安装

sudo apt-get install nvidia-driver-510-server

1.4验证安装

nvidia-smi

在这里插入图片描述

2.安装cuda

2.1查看显卡驱动版本对应的cuda版本
官网链接：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
在这里插入图片描述我NVIDIA-SMI 510.47.03下的cuda10.2版本，根据自己需求选择版本
2.2下载并安装cuda
官网链接：https://developer.nvidia.com/cuda-toolkit-archive
打开终端执行：

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

#下载完后执行：sudo sh cuda_10.2.89_440.33.01_linux.run
会有界面弹出，如果你在之前安装好了显卡驱动，会让你移除并重新安装，因此我没有使用这个指令安装，我使用的是这个指令：通过加参数，在安装的过程中可能就不会检测驱动，然后成功安装

sudo bash cuda_10.2.89_440.33.01_linux.run --toolkit --silent --override

2.3添加环境变量
打开~/.bashrc，根据自己的路径添加：

# cuda
export PATH="/usr/local/cuda-10.2/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH"

让配置生效：source ~/.bashrc
2.4验证安装

nvcc -V

在这里插入图片描述

3.安装cudnn

官网链接：https://developer.nvidia.com/rdp/cudnn-archive
3.1下载
一定要选择跟自己cuda版本匹配的！我选择的是cudnn8.1.0，点击下载
在这里插入图片描述 3.2安装

#解压
tar -xzvf  cudnn-10.2-linux-x64-v8.1.0.77.tgz
#会在当前文件夹下生成一个名为cuda的文件夹，将里面的文件拷贝到cuda的安装目录下
sudo cp  cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64/
sudo cp  cuda/include/cudnn* /usr/local/cuda-10.2/include/
#修改权限
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*

3.3验证安装

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

在这里插入图片描述 参考连接：
https://blog.csdn.net/weixin_43563233/article/details/114385130
https://blog.csdn.net/wenroudebaozi/article/details/114582502
https://shliang.blog.csdn.net/article/details/108994449