如果我需要在 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(使用前将#替换为@)