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

2023-12-24

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


  • 从以下位置下载每晚构建版本https://github.com/tensorflow/tensorflow#installation https://github.com/tensorflow/tensorflow#installation。 如何选择合适的构建:

    • 使用“Linux CPU-only”或“Linux GPU”,具体取决于您是否需要使用 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 to none。 Pip 解析该部分以检测 .whl 包是否适合平台,但该特定名称不适用于某些旧版本的 pip)

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

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

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

使用每晚 TensorFlow 构建通过 Cloud ML Engine 进行训练 的相关文章

随机推荐