Linux系统为 Ubuntu 18.04,显卡为GeForce RTX 3060 ,驱动版本号为515.57,安装CUDA 11.7。
下载CUDNN,官网地址。(需要注册)
进入到下载的目录下,进行解压:
tar -xf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz
将解压后的文件拷贝到CUDA对应的安装目录下,并更改文件属性:
sudo cp cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive/include/cudnn.h /usr/local/cuda-11.7/includesudo cp cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive/lib/libcudnn* /usr/local/cuda-11.7/lib64*sudo chmod a+r /usr/local/cuda-11.7/include/cudnn.h sudo chmod a+r /usr/local/cuda-11.7/lib64/libcudnn*
查看版本号:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
来源地址:https://blog.csdn.net/my__blog/article/details/125723725