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

【问题标题】：CUDA installation for TensorFlow. No cuDNN v4 for CUDA 7.5 at NVIDIATensorFlow 的 CUDA 安装。 NVIDIA 没有适用于 CUDA 7.5 的 cuDNN v4
【发布时间】：2016-08-30 07:36:16
【问题描述】：

我正在尝试安装 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？谢谢。

【问题讨论】：

您只需下载 7.0 的 cudnn v4 即可使用 7.5
7.5 可在developer.nvidia.com/cuda-downloads获得

标签： cuda tensorflow cudnn

【解决方案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/

【讨论】：

【解决方案2】：

没问题。我已经安装了 cuda 7.5 并在 ubuntu 16.04 上使用了 cuDNN v4。没有问题。

【讨论】：

【解决方案3】：

从头开始安装 CuDNN 很痛苦。

截至 2016 年 10 月，AWS 提供预装了 NVIDIA CUDA 7.5 的预构建 AMI。 Yipieee！

请看-

AWS Blog post

AWS AMI GPU Image

NVIDIA Digits4

【讨论】：

不知道为什么投反对票。我真的认为 AWS AMI 正在消除设置 CUDA 的许多痛苦步骤，这个答案将对未来的用户有所帮助。