发布于: Dec 11, 2017
此版本的 AWS Deep Learning AMI 支持 Apache MXNet 1.0,包括适用于 MXNet 的新模型服务功能,它只需几行代码,即可完成深度学习模型的打包、运行和提供。其中还包括全新的梯度压缩功能及全新的模型转换器,后者用于将在 Caffe 框架下编写的神经网络代码转换为 MXNet 代码,让开发人员更轻松地利用 MXNet 的可扩展性和性能。
这些 AMI 还为 NVIDIA Volta 提供改进后的框架支持。它们包含 PyTorch 0.3.0 版本,支持 NVIDIA CUDA 9 和 cuDNN 7,而且在 NVIDIA Volta GPU 上训练模型的性能得到显著提升。同时,它们还包含一个根据主实例构建,并与 NVIDIA 处理器合并来提供 Volta 支持的 TensorFlow 版本。此外,我们还在 AWS Deep Learning AMI 的 CUDA 9 版本中增加了 Keras 2.0 支持,与作为默认后端的 TensorFlow 配合使用。