发布于: Dec 13, 2018
适用于 Ubuntu 和 Amazon Linux 的 AWS Deep Learning AMI 现在提供以下更新版本的深度学习框架:PyTorch 1.0、MXNet 1.3.1 和 Chainer 5.1。该版本还包含 NVIDIA 堆栈升级,包括 CUDA 10、cuDNN 7.4.1 和 NCCL 2.3.7。对于基于 GPU 的培训,这些 AMI 中的 PyTorch 1.0 已经过完善配置,可充分利用 CUDA 10 中的性能改进。
此外,AWS Deep Learning AMI 还支持 TensorFlow、Keras、Gluon 和 Microsoft Cognitive Toolkit (CNTK) 等其他常见框架和接口,所有这些都已预先安装好并进行了完善配置,让您在几分钟内即可开始开发深度学习模型,同时充分利用 Amazon EC2 实例的计算能力。在您激活 Conda 环境时,Deep Learning AMI 会自动部署为您选择的 EC2 实例优化的更高性能版本的框架。如需 AWS Deep Learning AMI 支持的框架和版本的完整列表,请查看发行说明。
要快速开始使用 AWS Deep Learning AMI,请参阅开发人员指南中的入门指南和初级到高级教程。您还可以订阅我们的开发论坛,以获取发布公告并提出您的问题。