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对应的安装目录下,并更改文件属性:文章来源:https://www.toymoban.com/news/detail-679622.html
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*
查看版本号:文章来源地址https://www.toymoban.com/news/detail-679622.html
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
到了这里,关于linux 安装 CUDNN的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!