我一直在尝试使用以下步骤安装支持GPU的tensorflow:http://www.nvidia.com/object/gpu-accelerated-applications-tensorflow-installation.html 并使用: http://thelazylog.com/ 安装-tensorflow-与GPU的支持,上沙箱红帽/
当我尝试运行bazel build命令来构建tensorflow pip包时(设置了--config-cuda标志),这是我收到的错误消息:
The specified --crosstool_top '//third_party/gpus/crosstool:crosstool' is not a valid cc_toolchain_suite rule.
Run Code Online (Sandbox Code Playgroud)
奇怪的是,如果我删除--config = cuda标志,我在构建时没有收到错误消息,并且我能够成功安装tensorflow - 但没有GPU支持.
我使用 nvidia 说明时遇到了同样的问题。我所做的就是删除说明中的 git reset 行,它起作用了。
详细信息(来自错误消息):
关闭,重新打开终端
运行 git clone (再次),然后 cd tensorflow
运行./configure
Bazel 构建等
这可能是不相关的,但我遇到了 .whl 行的问题,错误消息是找不到轮子或类似的东西。这是“最后安装 TensorFlow pip 包”部分。为了解决我的情况,我在终端中一直输入“..._pkg/tensorflow”,然后按 Tab 键进行自动补全。弹出的文件名比指南中的文件名长得多,但它有效。另外,如果有人根据 nvidia 说明遇到 numpy notinstalled 消息,请将 python-pip 和 dev 替换为 python-numpy,然后再次运行该行进行安装。
配置:全新 Ubuntu 16.04、GTX970M、运行驱动程序 367.48(来自 CUDA 安装)、CUDA 8.0、CuDNN 5.1
完整安装路径:全新 Ubuntu,在安装过程中选择下载和第 3 方应用程序。控制面板 => 软件和更新 => 其他软件 => Canonical 勾选使用 CUDA 文档中的 nvidia 说明安装 CUDA,安装 .deb 格式的 CuDNN 5.1,其余部分来自 nvidia 链接。
我希望你一切都顺利!
(对于格式不佳,我深表歉意)
| 归档时间: |
|
| 查看次数: |
2416 次 |
| 最近记录: |