如何将cuDNN更新为更新版本?

时间:2016-07-01 05:42:55

标签: linux ubuntu nvidia caffe cudnn

cuDNN安装手册说

  

ALL PLATFORMS

     

将cuDNN存档解压缩到您选择的目录,如下所示。       然后按照以下平台特定说明进行操作。

     

LINUX

     

CD       export LD_LIBRARY_PATH =`pwd`:$ LD_LIBRARY_PATH

     

通过在编译中添加-I来添加到构建和链接过程       line和-L -lcudnn到你的链接行。

它似乎只是将pwd添加到LD_LIBRARY_PATH,所以我想只需更换pwd中的文件即可进行更新。

但似乎并不那么简单,因为在我做完这件事后,我收到了Theano的投诉说

  

cuDNN版本太旧了。更新到v5,是3007。

2 个答案:

答案 0 :(得分:8)

  1. 替换public void setLocale(String lang) { myLocale = new Locale(lang); Resources res = getResources(); DisplayMetrics dm = res.getDisplayMetrics(); Configuration conf = res.getConfiguration(); conf.locale = myLocale; res.updateConfiguration(conf, dm); Intent refresh = new Intent(this, YOURACTIVITY.class); refresh.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK | Intent.FLAG_ACTIVITY_CLEAR_TASK); startActivity(refresh);
  2. 中的cudnn.h
  3. 删除dir/cuda/include/
  4. 中的旧库文件
  5. 将新库文件添加到dir/cuda/lib64/

答案 1 :(得分:7)

我编写了一个脚本,可用于清理安装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

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