发布于: Jun 20, 2018
现在，Amazon SageMaker 经过预先配置，支持运行 PyTorch，从而扩大了目前可用的已经集成的现有 TensorFlow、Apache MXNet 和 Chainer 深度学习框架的阵容。此外，Amazon SageMaker 中的预配置 TensorFlow 容器现在支持版本 1.7 和 1.8。
在 Amazon SageMaker 中使用 PyTorch 与使用其他预构建的深度学习框架一样简单。由于 PyTorch 与 Python 深度集成，因此它允许您在网络中使用典型的 Python 控制流。此外，PyTorch 还支持动态计算图，可以更灵活地使用内存，并更好地支持递归计算。有关 PyTorch 的详细信息，请参阅此处的博文。
适用于 Amazon Sagemaker 的 PyTorch 预构建容器和 TensorFlow 1.7 与 1.8 现已在以下 AWS 区域推出：美国东部（弗吉尼亚北部）、美国东部（俄亥俄）、美国西部（俄勒冈）、欧洲（爱尔兰）和亚太地区（东京）。有关 PyTorch 的更多信息，请访问该文档。