tensorflow - SageMaker 和 TensorFlow 2.0

Question

使用 AWS Sagemeker 运行 TensorFlow 2.0 的最佳方式是什么？

截至今天(2019 年 8 月 7 日)，AWS 不提供 TensorFlow 2.0 SageMaker containers ，所以我的理解是我需要建立自己的。

最好使用的基本图像是什么？示例 Dockerfile？

Answer 1

编辑: Amazon SageMaker 现在支持 TF 2.0 及更高版本。

SageMaker + TensorFlow 文档:https://sagemaker.readthedocs.io/en/stable/frameworks/tensorflow/using_tf.html

支持的 Tensorflow 版本(和 Docker URI):https://aws.amazon.com/releasenotes/available-deep-learning-containers-images

原答案
这是一个使用 the underlying SageMaker Containers library 的 Dockerfile 示例(这是官方预构建的 Docker 镜像中使用的内容):

FROM tensorflow/tensorflow:2.0.0b1

RUN pip install sagemaker-containers

# Copies the training code inside the container
COPY train.py /opt/ml/code/train.py

# Defines train.py as script entrypoint
ENV SAGEMAKER_PROGRAM train.py

有关此方法的更多信息，请参阅 https://docs.aws.amazon.com/sagemaker/latest/dg/build-container-to-train-script-get-started.html