安装cuDNN-8.0

来源:互联网 发布:大众软件2014电子版 编辑:程序博客网 时间:2024/06/06 00:11

下载压缩文件

下载前,请确定cuDNN版本与cuda版本一致。cuDNN下载地址

下载前,请选择与深度学习框架匹配的cuDNN版本,详情参考cuDNN深度学习框架

准备压缩文件

解压下载好的压缩文件到当前用户根目录下。如下列出两款深度学习框架当前对应的压缩包文件名:

  • caffe---------->cudnn-8.0-linux-x64-v5.0-ga.tgz
  • tensorflow-->cudnn-8.0-linux-x64-v6.0.tgz

复制文件到cuda-8.0安装位置

sudo cp ~/cuda/include/cudnn.h /usr/local/cuda-8.0/includesudo cp ~/cuda/lib64/lib* /usr/local/cuda-8.0/lib64

创建连接

cd /usr/local/cuda-8.0/lib64#仅适用于cuDNN-8.0-v5.0sudo ln -sf libcudnn.so.5.0.5 libcudnn.so.5sudo ln -sf libcudnn.so.5 libcudnn.so#仅适用于cuDNN-8.0-v6.0sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6sudo ln -sf libcudnn.so.6 libcudnn.so
原创粉丝点击