下载地址 (需要注册)
https://developer.nvidia.com/rdp/cudnn-download
我的工作环境竟然是 CUDA10.1
,我只能选择这个:
下载下来后扔到服务器上:
cudnn-linux-x86_64-8.4.1.50_cuda10.2-archive.tar.xz
注意他是 xz 不是 gz 所以解压就是:
tar -xf cudnn-linux-x86_64-8.4.1.50_cuda10.2-archive.tar.xz
之前,是将 cudnn 的文件扔到 cuda 的路径就OK了,然而事实上,不一定放到cuda的路径,只要添加到 LD_LIBRARY_PATH
即可
解压之后,有目录cudnn-linux-x86_64-8.4.1.50_cuda10.2-archive
,其下有 include
和 lib
将该路径添加到 LD_LIBRARY_PATH
,给这个路径起一个名字:
cd cudnn-linux-x86_64-8.4.1.50_cuda10.2-archive
export CUDNN_DIR=$(pwd) # 给该路径起个名字
export LD_LIBRARY_PATH=$CUDNN_DIR/lib:$LD_LIBRARY_PAT # 注意有的可能是 lib64 而不是 lib
然而这样操作,添加的路径都是临时的,当然可以直接放到CUDA路径,也可以放到配置文件里:
echo "export CUDNN_DIR=$(pwd)" >> /etc/profile
echo "export LD_LIBRARY_PATH=$CUDNN_DIR/lib:$LD_LIBRARY_PATH" >> /etc/profile
之后刷新一下环境变量:
source /etc/profile
刷新完看看是否正确添加:
echo $CUDNN_DIR
/root/cudnn-linux-x86_64-8.4.1.50_cuda10.2-archive
返回这个就OK
再看看 LD_LIBRARY_PATH
对不对
echo $LD_LIBRARY_PATH
有这个就行