在装cudnn的时候,如果你参考的是这个链接
你会在安装cudnn 的章节中发现
查看cuDNN版本方法:
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
会出现如下报错:
cat: /usr/local/cuda-11.4/include/cudnn_version.h: 没有那个文件或目录
错误原因为:在执行
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.4/include
未把cudnn_version.h文件复制过去。我给的解决方案是执行下面代码:
sudo cp cuda/include/* /usr/local/cuda-11.4/include
sudo cp include/* /usr/local/cuda-11.5/include/
sudo chmod a+r /usr/local/cuda-11.5/lib64/*
sudo chmod a+r /usr/local/cuda-11.5/include/*
前两句意思是把所有的头文件都移植过去,以防未来需要。
执行
cat /usr/local/cuda-11.4/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
出现
(base) :~/下载/cudnn-11.4-linux-x64-v8.2.4.15$ cat /usr/local/cuda-11.4/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 2
#define CUDNN_PATCHLEVEL 4
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#endif /* CUDNN_VERSION_H */
即为成功