在Windows 10 x64上编译TensorFlow 1.4.0 GPU

时间:2017-10-24 23:11:38

标签: windows tensorflow tensorflow-gpu

似乎没有关于如何在Windows 10 x64上编译TensorFlow 1.4.0 GPU的详细文档。

我需要重新编译TF以为Windows 7 x64生产系统添加缺少的功能。

Installing TensorFlow from Sources上的官方Google链接只是说

  

我们不正式支持在Windows上构建TensorFlow;然而,   如果您不介意使用,可以尝试在Windows上构建TensorFlow   Windows上的高度实验性Bazel或TensorFlow CMake构建。

好的,我可以使用Bazel尝试编译像

这样的东西
bazel build -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-mfpmath=both --config=cuda -k //tensorflow/tools/pip_package:build_pip_package

但是如何将Python和NVIDIA依赖项合并到该构建中?

我应该使用CMake吗?

Windows https://pypi.python.org/pypi/tensorflow-gpu的轮子是如何编译的?

2 个答案:

答案 0 :(得分:1)

来自Adriano Carmezim的answer在github。有关在Windows上构建TensorFlow的完整说明,包括制作Windows Python轮子,请访问:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/cmake/README.md

github

解决了其他问题

答案 1 :(得分:0)

要在Windows上构建张量流,请使用cMake。