ubuntu安装cuda及cudnn简明教程

发布时间 2023-04-09 10:54:30作者: 吃旺旺雪饼的小男孩

前言

在上一篇文章https://www.cnblogs.com/tianyonglin/p/17285111.html中我们已经成功的安装了NVIDIA显卡驱动,本章给大家介绍一下如何安装cuda及cudnn,默认大家已经成功安装NVIDIA驱动。

安装cuda

1、查看自己的cuda_version最高支持

命令行输入nvidia-smi
image
可以看到cuda_version最高支持12.1

2、安装cuda

进入cuda官网https://developer.nvidia.com/cuda-toolkit-archive
image
选择适合自己的cuda版本,我这里选择11.6,因为后续我使用autoware需要使用cuda11.6
image
按照Base install指示进行安装即可,安装之后记得添加环境变量
export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc
恭喜你已成功安装cuda,输入nvcc -V验证一下吧
image

安装cudnn

进入cudnn官网https://developer.nvidia.com/rdp/cudnn-archive
image
在此之前记得注册nvidia账号
选择适合自己的版本,下载tar文件
image
参考官网安装教程https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html
image
命令行输入
cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2验证
得到以下输出
image
恭喜你,cudnn安装成功,版本号8.4.1
end!