Publicado: Mar 20, 2019
As AMIs do AWS Deep Learning agora estão disponíveis no Amazon Linux 2, a próxima geração do Amazon Linux, além do Amazon Linux e Ubuntu. Além disso, as AMIs do AWS Deep Learning agora vêm com o MXNet 1.4.0, o Chainer 5.3.0, o PyTorch 1.0.1 e o TensorFlow 1.13.1, que é personalizado diretamente da fonte e ajustado para treinamento de alto desempenho nas instâncias do Amazon EC2.
Em instâncias de CPU, o TensorFlow 1.13 é personalizado diretamente da fonte para acelerar o desempenho em processadores Intel Xeon Platinum que alimentam as instâncias EC2 C5. O treinamento de um modelo ResNet-50 com dados sintéticos do ImageNet usando os resultados da AMI do Deep Learning resultaram em uma taxa de transferência 9,4X mais rápida do que os binários padrão do TensorFlow 1.13. As instâncias vêm com uma compilação do TensorFlow 1.13 otimizada que está configurada com NVIDIA CUDA 10 e cuDNN 7.4, para aproveitar o treinamento de precisão mista nas GPUs Volta V100 que equipam as instâncias P3 do EC2. Para desenvolvedores que querem escalar o treinamento do TensorFlow para várias GPUs, as AMIs do Deep Learning vêm com a estrutura de treinamento distribuído do Horovod. A estrutura está totalmente otimizada para usar com eficiência as topologias compostas de cluster de treinamento distribuído das instâncias P3 do Amazon EC2. Treinar um modelo ResNet-50 usando o TensorFlow 1.13 e o Horovod na AMI do Deep Learning resulta em uma taxa de transferência 27% mais rápida do que o TensorFlow 1.13 em 8 nós.
As AMIs do AWS Deep Learning agora possuem a versão mais recente do Apache MXNet 1.4, que traz melhorias de desempenho e facilidade de uso. O MXNet 1.4 adiciona ligações Java para inferência, ligações Julia, operadores de fluxo de controle experimental, gerenciamento de memória JVM e muitos outros aprimoramentos ocultos. Esta versão também aprimora o suporte MXNet para Intel MKL-DNN com melhor otimização e quantização de gráficos. Esse recurso reduz o uso de memória e melhora o tempo de inferência sem uma perda significativa na precisão.
Comece a usar rapidamente as AMIs do AWS Deep Learning com os guias de conceitos básicos e os tutoriais para usuários iniciantes e avançados de nosso guia do desenvolvedor. Quando você ativa um ambiente do Conda, as AMIs do Deep Learning implantam automaticamente compilações das estruturas com maior performance, otimizadas para a instância do EC2 que você escolher. Para ver uma lista completa das estruturas e versões compatíveis com a AMI do AWS Deep Learning, consulte as notas de release. Você também pode assinar nosso fórum de discussão para obter anúncios de lançamentos e publicar suas perguntas.