2016-07-01 151 views
2

的cuDNN安装手册说如何将cuDNN更新为新版本?

所有平台

提取cuDNN文档到你选择的目录,以下称为。 然后按照以下平台特定的说明进行操作。

LINUX

CD 出口LD_LIBRARY_PATH =`pwd`:$ LD_LIBRARY_PATH

添加到您的构建和链接过程中加入-I到你的编译 线和-L -lcudnn你的链接线。

看来,它只是增加了pwdLD_LIBRARY_PATH,所以我想只是替换pwd文件将做更新。

但似乎并不像后,我已经这样做了这么简单,我从Theano说

cuDNN版本太旧投诉。更新到V5,是3007

回答

6
  1. dir/cuda/include/
  2. 更换cudnn.h删除旧的库文件dir/cuda/lib64/
  3. 添加新的库文件dir/cuda/lib64/
+1

还要确保您的系统路径设置正确。假设'packages /'在你的'CWD'中,是否为 –

2

我写了一个脚本里面用于清理安装cuDNN版本或将现有的cuDNN更改为更旧/更新版本。您可以从下载脚本:

https://github.com/dnzzcn/cuDNNv

这是脚本的作用:

#!/bin/bash 

rm -f /usr/include/cudnn.h 
rm -f /usr/lib/x86_64-linux-gnu/*libcudnn* 
rm -f /usr/local/cuda-*/lib64/*libcudnn* 


cp -P packages/cudnn/include/cudnn.h /usr/include 
cp -P packages/cudnn/lib64/libcudnn* /usr/lib/x86_64-linux-gnu/ 
chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn* 

rm -rf packages/cudnn 

它您需要的版本可以自动执行安装操作。

+0

?还需要'sudo'? – Shai

+1

是的,packages文件夹包含在版本库目录中。有一个python脚本调用上面的bash脚本来获得所需的cuDNN版本;虽然我不确定,但它应该没有sudo。 – dnzzcn

相关问题