google-cloud-ml - 使用每晚 TensorFlow 构建通过 Cloud ML Engine 进行训练

Question

如果我需要在 Cloud ML Engine 训练作业中使用夜间 TensorFlow 构建，我该怎么做？

Answer 1

• 从 https://github.com/tensorflow/tensorflow#installation 下载夜间构建.如何选择正确的构建:

  • 根据您是否需要使用 GPU 进行训练，使用“仅 Linux CPU”或“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

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

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

请注意，除了使用夜间构建，您还可以按照 https://www.tensorflow.org/install/install_sources 中所述从源代码构建 TensorFlow。 .