Have used this digits ami on aws in the past for caffe and torch.

https://aws.amazon.com/marketplace/pp/B01DJ93C7Q/ref=srh_res...