3 ms·
Is there an AMI that comes with Tensorflow/keras with GPU support preinstalled or you have to do it yourself?
by sethgecko 9y ago
Is there an AMI that comes with Tensorflow/keras with GPU support preinstalled or you have to do it yourself?
- Smerity 9y agoAmazon offer an official AMI which comes preloaded with various deep learning frameworks: MXNet, TensorFlow, CNTK, Caffe/2, Theano, Torch and Keras. For the P3 (Volta V100) instances you'll want to ensure you use an AMI preloaded with CUDA 9, though not all DL frameworks are happy with that yet. https://aws.amazon.com/amazon-ai/amis/ https://aws.amazon.com/amazon-ai/amis/
- mv4 9y agoDidn't know they had an AMI like that. Thank you.
- sipherhex 9y agoBe careful with the non-CUDA 9 AMIs. CUDA 8 programs will run, but terribly slowly as they JIT their GPU code without optimization for Volta. You want the CUDA 9 AMI version (https://aws.amazon.com/marketplace/pp/B076TGJHY1?qid=1509090457271 https://aws.amazon.com/marketplace/pp/B076TGJHY1?qid=1509090...), but it currently only has MXNet and TF. If you need other frameworks there's the NVIDIA AMI (https://aws.amazon.com/marketplace/pp/B076K31M1S?qid=1509090567000 https://aws.amazon.com/marketplace/pp/B076K31M1S?qid=1509090...) and Volta optimized containers for NVCaffe, Caffe2, CNTK, Digits, MXNet, PyTorch, TensorFlow, Theano, Torch, CUDA 9/CuDNN7/NCCL.