参考链接:Stack overflow: Installing CUDNN for general usage without root access
下面是机翻内容:
没有 root access,安装cudnn
可以使用安装在服务器中的CUDA的CUDNN,这是我所做的工作。 首先,您只需在当地的空间中简单地进行文件:
Home/local
并使它包含包含和lib文件夹(我猜大多数都有这些本地文件夹)。
Home/local/include
Home/local/lib
然后下载CUDNN并将CUDNN文件夹中的 include
和lib64
移动到本地include
和lib
文件夹中的内容(您刚刚制作)
最后,将这两个环境路径添加到您的 .bashrc
文件
export CPATH=$CPATH: $ HOME/local/include #
$HOME
是指你存放位置的前缀 ,命令不含空格export LD_LIBRARY_PATH=$LD_LIBRARY_PATH: $ HOME/local/lib
这样它就会工作了。
顺便说一句,如果在成功安装CUDNN后遇到“忘记内存”问题,请在运行代码之前在终端中输入此行:
export CUDA_VISIBLE_DEVICES=0
查看cudnn版本信息
旧版查看版本信息在cudnn.h
, 新版本的版本信息在 cudnn_version.h
,例如cudnn_8.1.0版本。
# ${CUDNN_H_PATH} 举例:/usr/local/cuda-10.0/include/cudnn.h
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2`
运行结果如下:
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
那么该cudnn版本就为===>7.6.2