5

cuDNN 安装手册说

所有平台

将 cuDNN 存档解压缩到您选择的目录，以下称为 . 然后按照以下特定于平台的说明进行操作。

LINUX

cd 导出 LD_LIBRARY_PATH=`pwd`:$LD_LIBRARY_PATH

通过将 -I 添加到您的编译行并将 -L -lcudnn 添加到您的链接行来添加到您的构建和链接过程。

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

但这似乎并不那么简单，因为在我完成此操作后，我收到了 Theano 的投诉说

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

4

3 回答 3

10
  1. 替换cudnn.h_dir/cuda/include/
  2. 删除旧的库文件dir/cuda/lib64/
  3. 将新的库文件添加到dir/cuda/lib64/
于 2016-07-01T08:52:34.550 回答
9

我编写了一个脚本，可用于全新安装 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

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

于 2017-09-11T20:39:10.973 回答
7

这是一个老问题，但现在我们有了 conda。我们不再需要手动替换这些文件。conda install cudnn做所有的伎俩。

于 2020-01-02T02:51:40.893 回答