【发布时间】:2017-02-09 08:35:33
【问题描述】:
我一直在尝试使用以下步骤安装支持 GPU 的 tensorflow: http://www.nvidia.com/object/gpu-accelerated-applications-tensorflow-installation.html 并且还使用: http://thelazylog.com/install-tensorflow-with-gpu-support-on-sandbox-redhat/
这是我在尝试运行 bazel build 命令来构建 tensorflow pip 包(设置了 --config-cuda 标志)时收到的错误消息:
The specified --crosstool_top '//third_party/gpus/crosstool:crosstool' is not a valid cc_toolchain_suite rule.
奇怪的是,如果我删除 --config=cuda 标志,我在构建时不会收到错误消息,并且我能够成功安装 tensorflow - 但没有 GPU 支持。
【问题讨论】:
-
您使用的是哪个 CUDA 版本? 8.0还是有点问题的。你也可以在 Github issues for tensor flow 中找到答案。
-
谢谢。是的,我正在使用 cuda 8.0。也许我应该尝试旧版本。这是我在 github 上找到的最接近的:github.com/tensorflow/tensorflow/issues/4368 我尝试使用旧版本(0.2.2)的 bazel,就像人们在此处建议的那样,但仍然遇到相同的错误。
标签: tensorflow bazel