作者:xueer2502938817_416 | 来源:互联网 | 2022-10-16 15:58
使用AWS Sagemeker运行TensorFlow 2.0的最佳方法是什么?
截至今天(2019年8月7日),AWS不提供TensorFlow 2.0 SageMaker容器,因此我的理解是我需要构建自己的容器。
什么是最好的基本映像使用?示例Dockerfile?
1> 小智..:
这是一个使用基础SageMaker Containers库的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