3

如果我需要在 Cloud ML Engine 训练作业中使用每晚的 TensorFlow 构建,我该怎么做?

4

1 回答 1

5
  • 从https://github.com/tensorflow/tensorflow#installation下载每晚构建。如何选择正确的构建:

    • 根据您是否需要使用 GPU 进行训练,使用“Linux CPU-only”或“Linux GPU”,
    • 使用 Python 2 构建。
  • 重命名 .whl 文件,例如

    mv tensorflow-1.0.1-cp27-cp27mu-linux_x86_64.whl \
       tensorflow-1.0.1-cp27-none-linux_x86_64.whl
    

    (这里我们将 .pip 重命名为.pip 解析该部分cpu27munone检测 .whl 包是否适合某个平台,但该特定名称不适用于某些旧版本的 pip)

  • 将 .whl 文件上传到 GCS,并package_uris 在提交 Cloud ML Engine 训练作业时将其指定为其中之一。

请注意,您还可以按照https://www.tensorflow.org/install/install_sources中所述从源代码构建 TensorFlow,而不是使用每晚构建。

于 2017-03-31T20:17:45.100 回答