Windows上的Tensorflow自定义编译-Java 学习之路

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

我've followed the instructions of other articles, both on 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计算工具包/ CUDA / v8.0 / include / cudnn.h

我已经仔细检查过，那个文件确实存在，所以我不确定为什么我会收到这个错误 .

编辑：也试图通过Powershell运行，达到了同样的目的 .

任何帮助将非常感激 .

2 回答

3
尝试在Windows上构建Tensorflow时，我遇到了完全相同的错误（使用cuDNN 5.1） . 我通过从msys2终端启动bazel（而不是从windows命令提示符）并在尝试构建之前手动设置BAZEL_SH环境变量来修复它 .
```
export BAZEL_SH=c:/tools/msys64/usr/bin/bash.exe
bazel build -c opt --config=win-cuda tensorflow/cc:cc_ops
```
回复于 2024-05-01T16:33:27+08:00
0
以下步骤帮助我在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：无法打开编译器生成的文件：''：无效的参数Target // tensorflow / tools / pip_package：build_pip_package无法构建
回复于 2024-05-01T16:33:27+08:00

Windows上的Tensorflow自定义编译

2 回答

相关问题