需要TensorFlow依赖项。如何在Windows

时间:2015-11-19 17:01:21

标签: windows gpu tensorflow

我有兴趣让TensorFlow在Windows上运行,但是目前我意识到这是不可能的,因为某些依赖项无法用于Windows,例如巴泽勒。

需要出现,因为我目前了解它,从TensorFlow访问GPU的唯一方法是通过非虚拟安装Linux。我确实知道我可以双启动进入Linux安装,但更愿意避免这种做法。

要解决这个问题,我需要整个依赖关系链来构建TensorFlow,因为想知道这是否已经存在。

我也意识到,从源代码构建时,我可以捕获构建输出作为一个可靠的开始,但是如果它已经知道,我希望避免这种工作。

有一个在Windows上运行的Bazel测试版 - https://github.com/dslomov/bazel-windows

请参阅相关的GitHub问题以在Windows上运行TensorFlow。 - https://github.com/tensorflow/tensorflow/issues/17

在Windows上运行的另一个原因是可以移植到Xbox One。

我找到了可能的答案,仍然需要检查它。这将生成一个依赖图作为点文件。

$ bazel query 'deps(//tensorflow/tools/pip_package:build_pip_package)' --output graph > tensorflow.dependency.dot

3 个答案:

答案 0 :(得分:4)

现在有三种在Windows上构建和/或运行TensorFlow的主要选项:

  • 您可以从PyPI安装支持GPU的TensorFlow 0.12rc0 PIP包:pip install tensorflow-gpu
  • 您可以使用experimental CMake build自行构建支持GPU的PIP包。这也使您能够在Visual Studio中使用TensorFlow。可以找到此版本的文档here
  • 初步支持使用Bazel for Windows构建TensorFlow。但是,我们仍在使用此版本解决一些错误。

答案 1 :(得分:0)

这可能不是你想要的一种方式在Windows下运行TensorFlow是安装一个虚拟机(VMWare播放器v12可以免费用于非商业用途),然后在其中安装Ubuntu,最后在Ubuntu中安装TensorFlow。适合我。

答案 2 :(得分:0)

自2017年年初以来,Tensorflow现已成为officially supported on Windows and can be installed via pip

pip install --upgrade tensorflow
pip install --upgrade tensorflow-gpu

或直接提取包(选择符合您需求的包,例如x64 / gpu)

# x86 / CPU
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.0.0-cp35-cp35m-win_x86_64.whl

# x64 / CPU
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.0.0-cp35-cp35m-win_amd64.whl
# x64 / GPU
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.0.0-cp35-cp35m-win_amd64.whl