下载CUDA11.8

下载CUDA11.8

选择对应的系统 架构 OS 版本

逐步执行上图命令

编辑环境变量文件

sudo gedit ~/.bashrc

配置环境变量

export PATH=/usr/local/cuda-11.8/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH

重启
重启
重启
重要的事情说三边 查看版本

nvcc -V

结果

安装cudnn

下载cudnn

找到适合你的cudnn
https://developer.nvidia.com/rdp/cudnn-arcHive

安装cudnn

下载好以后解压(注意核对下载和解压名称是否一致）
tar -zxvf cudnn-11.8-linux-x64-v7.1.tgz

解压后进到里面：只在这一层级，右击打开终端，一条一条输入以下命令

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/sudo chmod a+r /usr/local/cuda/include/cudnn.hsudo chmod a+r /usr/local/cuda/lib64/libcudnn*

查看装好了吗？

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

最后

重启
重启
重启 重要的事情说三边

输入

nvidia-smi

来源地址：https://blog.csdn.net/FYH666520/article/details/127701315