TensorFlow的CUDA安装。NVIDIA没有适用于CUDA 7.5的cuDNN v4。

Question

我正在尝试安装CUDA（用于TensorFlow）。建议使用cuDNN v.4和CUDA 7.5。但在https://developer.nvidia.com/rdp/cudnn-download上，只提供了适用于CUDA 7.0的cuDNN v.4和适用于CUDA 7.5的cuDNN v.5（发行版本）。我该在哪里获取适用于CUDA 7.5的cuDNN v.4？谢谢。

- Eugene Livshitz

---

你只需下载适用于7.0的cudnn v4版本，它也能在7.5上运行。 - fabrizioM

7.5 可在 https://developer.nvidia.com/cuda-downloads 下载。 - gliptak

Answer 1

正如@fabrizioM所指出的那样，您可以使用cuDNN v.4适用于CUDA 7.0。我也花了很多时间安装CUDA/cuDNN。

供参考，https://no2147483647.wordpress.com/2016/01/16/setup-amazon-aws-gpu-instance-with-mxnet/有很好的说明：

tar -zxf cudnn-7.0-linux-x64-v4.0-rc.tgz 
cd cuda
sudo cp lib64/* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/

- Sung Kim

Answer 2

没问题。我已经在Ubuntu 16.04上安装了CUDA 7.5，并使用cuDNN v4。这方面没有问题。

- mrphoenix13

Answer 3

从头安装CuDNN很痛苦。

截至2016年10月，AWS提供了预先安装了NVIDIA CUDA 7.5的AMI。太好了！

请参见 -

AWS博客文章

AWS AMI GPU镜像

NVIDIA Digits4

- Srikar Appalaraju

---

不确定为什么会被踩。我真的认为AWS AMI正在消除设置CUDA的许多痛苦步骤，这个答案将有助于未来的用户。 - Srikar Appalaraju