【发布时间】: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