ubuntu安装cuda及cudnn简明教程

前言

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

安装cuda

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

命令行输入nvidia-smi

可以看到cuda_version最高支持12.1

2、安装cuda

进入cuda官网https://developer.nvidia.com/cuda-toolkit-archive

选择适合自己的cuda版本，我这里选择11.6，因为后续我使用autoware需要使用cuda11.6

按照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验证一下吧

安装cudnn

进入cudnn官网https://developer.nvidia.com/rdp/cudnn-archive

在此之前记得注册nvidia账号
选择适合自己的版本，下载tar文件

参考官网安装教程https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

命令行输入
cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2验证
得到以下输出

恭喜你，cudnn安装成功，版本号8.4.1
end！