当前位置: 首页>>技术问答>>正文


我如何在Ubuntu 16.04上安装CuDNN?

, , ,

问题描述

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

最佳解决方案

第0步:从标准存储库安装cuda。 (见How can I install CUDA on Ubuntu 16.04?)

第1步:注册一个nvidia开发者帐号和download cudnn here(约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*

次佳解决方案

从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 site,然后逐一安装。

 sudo dpkg -i <library_name>.deb

第三种解决方案

  1. 在NVidia的网站上注册。在他们获得您的帐户批准之前,可能需要一两天时间。至少在我注册时曾经是这种情况。

  2. Download and Install latest CUDA from NVidia或适用于您将使用的软件(如果有的话)的最新版本,在此情况下为您的T-Flow版本。请注意,通过点击安装通过ubuntu的标准包管理器可能无法正常工作。相反,您可能必须在终端中遵循these instructions才能安装.deb包。之后,您必须将add a few lines添加到.bashrc,或者在您的情况下适用。例如,如果您正在配置服务器,那么它可能会是一个不同的地方,也许是在您的应用程序自动启动之前的某个地方,因为在这种情况下.bashrc可能无法执行。

  3. Download CuDNN from NVidia我使用了“Linux for Library”版本,对.deb软件包没有太大的好运。

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

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

第四种方案

另外,你可以下载基于Debian的发行版的deb软件包。

在NVIDIA网页上,开发者档案可用于下一个文件:

  • 适用于Linux的cuDNN v5.1运行时库(Deb)

  • 适用于Linux的cuDNN v5.1开发库(Deb)

  • cuDNN v5.1代码示例和用户指南Linux(Deb)

我测试了这个,通过我的机器与Debian(Stretch)和TensorFlow工作!

参考资料

本文由Ubuntu问答整理, 博文地址: https://ubuntuqa.com/article/880.html,未经允许,请勿转载。