Distributed training on AWS deep learning AMI 9.0