☰

如何在 Ubuntu 16.04 上安装 CuDNN？

tag-icon • tag-icon 16.04 nvidia cuda gpu
如何在 Ubuntu 16.04 上安装 CuDNN？

对于 TensorFlow，我想安装 cuda 和 CuDNN。如何在 Ubuntu 16.04 上执行此操作？

答案1

步骤 0：从标准存储库安装 cuda。（参见如何在 Ubuntu 16.04 上安装 CUDA？）

步骤 1：注册 nvidia 开发者账户并点击此处下载 cudnn(大约 80 MB)

步骤 2：检查您的 cuda 安装在哪里。对于从存储库安装，它是/usr/lib/...和/usr/include。否则，它将是/usr/local/cuda/或。您可以使用或/usr/local/cuda-<version>进行检查which nvccldconfig -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*

运行文件安装：

$ 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*

答案2

从 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.deb从以下位置下载nvidia 网站 并按照下列方式逐一安装。

 sudo dpkg -i <library_name>.deb

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

答案3

  1. 在 NVidia 网站上注册。他们可能需要一两天时间才能批准您的帐户。至少我注册时是这样的。

  2. 从 NVidia 下载并安装最新的 CUDA，或者适合您使用的软件的最新版本（如果有），在本例中是您的 T-Flow 版本。

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

    相反，你可能需要遵循这些说明在终端安装.deb包。之后你必须添加几行到.bashrc，或根据您的情况在任何适当的位置。例如，如果您正在配置服务器，它可能会位于其他位置，可能是应用程序自动启动之前的某个位置，因为.bashrc在这种情况下可能不会执行。

  3. 从 NVidia 下载 CuDNN

    我使用了“Linux 库”版本，但软件包方面运气不佳.deb。

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

  5. 打开 CuDNN 存档并将适当内容复制到 CUDA 安装文件夹 (cuda/lib64/和cuda/include/) 内的适当位置。我通常sudo nautilus从那里直观地进行操作。

答案4

另外，您还可以下载基于 Debian 的发行版的 deb 包。

在 NVIDIA 网页上，开发人员配置文件可用以下文件：

  • 适用于 Linux 的 cuDNN v5.1 运行时库 (Deb)
  • 适用于 Linux 的 cuDNN v5.1 开发库 (Deb)
  • cuDNN v5.1 代码示例和用户指南 Linux (Deb)

我在我的 Debian（Stretch）机器上测试了这一点，TensorFlow 可以正常运行！

相关内容