如何在 Ubuntu 16.04 上安装 CuDNN?

Mar*_*oma 112 nvidia cuda gpu 16.04

对于 TensorFlow,我想安装 cuda 和 CuDNN。我如何在 Ubuntu 16.04 上做到这一点?

Mar*_*oma 148

第 0 步:从标准存储库安装 cuda。(请参阅如何在 Ubuntu 16.04 上安装 CUDA?

第一步:注册一个nvidia开发者账号,在这里下载cudnn(大约80MB

第 2 步:检查您的 cuda 安装位置。对于从存储库安装,它是/usr/lib/.../usr/include。否则,它将是/usr/local/cuda/or /usr/local/cuda-<version>。你可以用which nvcc或检查它ldconfig -p | grep cuda

第 3 步:复制文件:

存储库安装:

$ cd folder/extracted/contents
$ sudo cp -P include/cudnn.h /usr/include
$ sudo cp -P lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
$ sudo chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*
Run Code Online (Sandbox Code Playgroud)

运行文件安装:

$ cd folder/extracted/contents
$ sudo cp include/cudnn.h /usr/local/cuda/include
$ sudo cp lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
Run Code Online (Sandbox Code Playgroud)

  • 为了让 Tensorflow 找到所有内容,我必须将 `include/cudnn.h` 和 `lib64/` 中的库复制到 `/usr/local/cuda-8.0/include` 和 `/usr/local/cuda-8.0/lib64 `(使用 CUDA 8.0、Ubuntu 14.04、Tensorflow 0.12.0rc0) - 也许这对某人有帮助。 (38认同)
  • 添加`-P`保留符号链接,即`sudo cp -P lib64/libcudnn* /usr/lib/x86_64-linux-gnu/`,并避免出现以下消息:`/sbin/ldconfig.real: /usr/lib /x86_64-linux-gnu/libcudnn.so.5 不是符号链接` (15认同)

GPr*_*hap 44

从 5.1 开始,您无法根据@Martin 提到的内容进行安装。libcudnn6_6.0.21-1+cuda8.0_amd64.deb, libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb, libcudnn6-doc_6.0.21-1+cuda8.0_amd64.debnvidia网站下载 并按照以下方式一一安装。

 sudo dpkg -i <library_name>.deb
Run Code Online (Sandbox Code Playgroud)

编辑:您必须首先安装运行时(libcudnn6_6.0.21-1+cuda8.0_amd64.deb),因为开发依赖于运行时(感谢@tinmarino)

  • 从源码编译Tensorflow的时候知道cuDNN库安装路径是`/usr/lib/x86_64-linux-gnu/`就好了 (9认同)
  • 你必须先安装 `runtime` 因为 `dev` 依赖它 (3认同)

Ího*_* Mé 13

  1. 在 NVidia 的网站上注册。他们可能需要一两天时间才能批准您的帐户。至少在我注册时曾经是这样。
  2. 从 NVidia 下载并安装最新的 CUDA,或适合您将使用的软件的最新版本(如果有),在这种情况下是您的 T-Flow 版本。

    请注意,通过单击通过 ubuntu 的标准包管理器进行安装可能无法正常工作。

    相反,您可能必须按照终端中的这些说明安装包.deb。在此之后,你就必须添加几行,以.bashrc你的情况适合,或其它地方。例如,如果你正在配置一个服务器,它可能会在一个不同的地方,可能在你的应用程序自动启动之前的某个地方,因为.bashrc在这种情况下可能不会被执行。

  3. 从 NVidia 下载 CuDNN

    我使用了“Library for Linux”版本,对.deb软件包没有太大的帮助。

  4. 您可以通过 找到 CUDA 所在的位置 which nvcc。通常/usr/local/cuda/是指向您当前安装版本的符号链接。

  5. 打开 CuDNN 存档并将适当的内容复制到 CUDA 安装文件夹(cuda/lib64/cuda/include/)中的适当位置。我通常sudo nautilus从视觉上从那里做。


小智 8

快进 2018 年,NVIDIA 现在提供 cuDNN 7.x 供下载。安装步骤仍然与@GPrathap 描述的步骤相似。但是如果你想用较新的版本替换旧的 cuDNN 版本,你需要在安装之前先将其删除。

回顾一下:

步骤 0. 验证您是否已经安装了 CUDA 工具包。如果还没有,请继续安装 CUDA 工具包。

步骤 1. 前往 NVIDIA 开发者门户https://developer.nvidia.com/cudnn并下载 cuDNN。

步骤 2. 如果您之前安装了 cuDNN,请将其删除

sudo dpkg -r <old-cudnn-runtime>.deb
sudo dpkg -r <old-cudnn-dev>.deb
Run Code Online (Sandbox Code Playgroud)

步骤 3. 使用 dpkg 安装 cuDNN 库(runtime、dev、doc)

sudo dpkg -i <new-cudnn-runtime>.deb
sudo dpkg -i <new-cudnn-dev>.deb
sudo ldconfig
Run Code Online (Sandbox Code Playgroud)

第 4 步。如果要查找库的安装位置,可以更新定位索引,然后找到库位置。

sudo updatedb
locate libcudnn
Run Code Online (Sandbox Code Playgroud)

如果您专门针对 CUDA 工具包 9.1 安装 cuDNN 7.x,本文提供了更多详细说明,可能会有所帮助:http : //tech.amikelive.com/node-679/quick-tip-installing-cuda-deep-神经网络7-cudnn-7-x-library-for-cuda-toolkit-9-1-on-ubuntu-16-04/