python - 在 Sagemaker 上使用 TensorFlow 的 Python 3

Question

我了解 Sagemaker 目前不支持带有 Tensorflow 的 Python 3（根据此https://github.com/aws/sagemaker-python-sdk/issues/19）

但是是否可以像这里解释的那样使用 Python 3 和 Tensorflow 创建自己的 docker 容器？ https://github.com/awslabs/amazon-sagemaker-examples/blob/master/advanced_functionality/scikit_bring_your_own/scikit_bring_your_own.ipynb

Answer 1

是的，如果您创建自己的 Docker 容器，则可以将 Tensorflow 与 Python 3 一起使用。默认的 Tensorflow Dockerfile 可在此处获得，您可以根据需要对其进行修改。

Answer 2

新的脚本模式容器支持带有框架版本 1.11 的 python 3。AWS 支持给我的印象是旧的 tf.estimator 样式容器不支持 python 3。