ubuntu - 是否可以为 GTX 1070 构建 TensorFlow？

Question

我有一个 Ubuntu 14.04 LTS 64 位，带有 Nvidia 显卡 - GTX 1070(第 10 代)。我正在尝试构建 TensorFlow。

我尝试使用 CUDA 7.5 和 CuDNN 5 构建它，但结果发现我安装的 CUDA 7.5 需要 352.63.0 视频驱动程序，而我从 Nvidia 下载的 GTX 1070 视频驱动程序是 367.35 - 一个较新版本。

TensorFlow 成功构建，但是当我运行示例时，运行时出现问题:

boyko@boyko-pc:~/Desktop/tensorflow/tensorflow/models/image/mnist$ LD_LIBRARY_PATH=/usr/local/cuda-7.5/targets/x86_64-linux/lib python3 convolutional.py

由于驱动程序不匹配，未能找到 CUDA:

E tensorflow/stream_executor/cuda/cuda_driver.cc:491] failed call to cuInit: CUDA_ERROR_NO_DEVICEE tensorflow/stream_executor/cuda/cuda_diagnostics.cc:296] kernel version 367.35.0 does not match DSO version 352.63.0 -- cannot find working devices in this configuration

完整日志 - http://pastebin.com/xiYtNsHk

CUDA 7.5 需要驱动 352.63 视频驱动程序，但 GTX 1070 需要 367.35。问题是TensorFlow官方仅支持CUDA 7.5。所以要求有点矛盾。

我需要做什么？是否可以在 GTX 1070 上使用 352.63 驱动程序，即使它启用了有限的功能集，它也能运行吗？或者是否有针对该驱动程序构建的 CUDA 7.5 版本，或者是否有针对 CUDA 8.0 构建 TensorFlow 的方法？

这是我发现的一个相关问题 - Tensorflow Bazel 0.3.0 build CUDA 8.0 GTX 1070 fails .

Answer 1

您需要加入NVIDIA developer program为了立即获得 CUDA 8.0 候选版本，因为正式版本尚未发布。你应该能够在你的 1070 上安装它(我已经在 1080 上运行了)。旧的驱动程序和 CUDA 版本甚至不支持 Pascal 架构，因此您必须走这条路。

然后你需要从source构建tensorflow ，这次应该可以了。