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/include
sudo 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

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐