【问题标题】:Tensorflow Custom Compile on WindowsWindows 上的 TensorFlow 自定义编译
【发布时间】:2017-08-30 13:53:24
【问题描述】:

所以,我已经通过 Chocolatey 安装了 Bazel,安装了 Python 3.5 和 2.7,安装了 CUDA v8 和 cuDNN v6,并安装了 JDK 8.0,我现在正尝试在我的 Windows 10 设备上使用 AVX 自定义构建 TensorFlow 、AVX 2 和 CUDA。 TensorFlow-GPU,预建版本,确实可以工作,我已经测试并成功运行了。

我已按照其他文章的说明进行操作,包括 TensorFlows 的实际站点(尝试改编 Linux/Mac 安装中的某些部分)和此处。我做的最远的是;通过 Msys2 克隆 Github 存储库,运行 configure.py,然后尝试通过 bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package 构建我收到一个错误,其标题是:

读取 java.io.IOException 时出错:CreateProcess():系统找不到指定的文件。 : C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v8.0/include/cudnn.h

我已经仔细检查过,该文件确实存在,所以我不确定为什么会出现此错误。

编辑:也尝试通过 Powershell 运行,达到了相同的点。

任何帮助将不胜感激。

【问题讨论】:

  • 我在使用 cuDNN v6 时遇到了问题。改用 v5.1 试试看是否可行。
  • 如果 Msys2 不起作用,我会尝试一下,谢谢。

标签: tensorflow compilation bazel cudnn


【解决方案1】:

我在尝试在 Windows 上构建 Tensorflow(使用 cuDNN 5.1)时遇到了完全相同的错误。我通过从 msys2 终端(而不是从 Windows 命令提示符)启动 bazel 并在尝试构建之前手动设置 BAZEL_SH 环境变量来修复它。

export BAZEL_SH=c:/tools/msys64/usr/bin/bash.exe
bazel build -c opt --config=win-cuda tensorflow/cc:cc_ops

【讨论】:

  • 它显示“bash: bazel: command not found”。有什么想法吗?
  • 似乎 bazel 不在您的 bash 路径中。 which bazel 说什么?
  • which: no bazel in (/usr/local/bin:/usr/bin:/bin:/opt/bin:/c/Windows/System32:/c/Windows:/c/Windows/System32/Wbem:/c/Windows/System32/WindowsPowerShell/v1.0/:/usr/bin/site_perl:/usr/bin/vendor_perl:/usr/bin/core_perl)@Mitya
【解决方案2】:

以下步骤帮助我在 Windows 10 上编译 Tensorflow

pacman -Syuu patch
ln -s "c:\python27\python.exe" /usr/bin/python

export BAZEL_SH=c:/tools/msys64/usr/bin/bash.exe
"C:\Documents and Settings\All Users\chocolatey\bin\bazel.exe" build --config=opt --config=win-cuda //tensorflow/tools/pip_package:build_pip_package

但经过 1 小时的编译后,我又遇到了一个错误:

C:\tools\msys64\tmp_bazel_dmitry\x1e5egqw\execroot\org_tensorflow\external\protobuf_archive\python\google\protobuf\internal\api_implementation.cc : 致命错误 C1083: 无法打开编译器生成的文件: '': 无效 参数目标 //tensorflow/tools/pip_package:build_pip_package 构建失败

【讨论】:

    猜你喜欢
    • 2017-11-02
    • 1970-01-01
    • 2018-11-28
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多