问题描述
对于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 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*
次佳解决方案
从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
第三种解决方案
-
在NVidia的网站上注册。在他们获得您的帐户批准之前,可能需要一两天时间。至少在我注册时曾经是这种情况。
-
Download and Install latest CUDA from NVidia或适用于您将使用的软件(如果有的话)的最新版本,在此情况下为您的T-Flow版本。请注意,通过点击安装通过ubuntu的标准包管理器可能无法正常工作。相反,您可能必须在终端中遵循these instructions才能安装
.deb
包。之后,您必须将add a few lines添加到.bashrc
,或者在您的情况下适用。例如,如果您正在配置服务器,那么它可能会是一个不同的地方,也许是在您的应用程序自动启动之前的某个地方,因为在这种情况下.bashrc
可能无法执行。 -
Download CuDNN from NVidia我使用了“Linux for Library”版本,对
.deb
软件包没有太大的好运。 -
您可以通过
which nvcc
找到CUDA的位置。通常/usr/local/cuda/
将成为您当前安装版本的符号链接。 -
打开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工作!