Need an AMI option that automatically installs latest versions of CUDA and deep learning frameworks

aws-samples / aws-deep-learning-ami-ubuntu-dcv-desktop

MIT No Attribution

15 stars 2 forks source link

Need an AMI option that automatically installs latest versions of CUDA and deep learning frameworks #5

Closed ajayvohra2005 closed 2 years ago

ajayvohra2005 commented 2 years ago

The Deep Learning AMI has fixed versions of CUDA , Tensorflow, PyTorch, and MxNet. Need an AMI option in the CloudFormation template that starts with Ubuntu Pro AMI from AWS and automatically installs the latest version of CUDA , Tensorflow, PyTorch, and MxNet.

ajayvohra2005 commented 2 years ago

Hash commit f938bce7f9389325252cb6c9685d59beaf42a1d8 resolves this issue.

Added a CloudFormation template parameter to select the AMI type: AWS Deep Learning AMI (Ubuntu 18.04), or Ubuntu Pro 20.04 LTS. If Ubuntu Pro is selected, CUDA, Tensorflow, Pytorch, and MxNet are installed via cloud init script.