spotty-cloud / spotty

Training deep learning models on AWS and GCP instances
https://spotty.cloud
MIT License
491 stars 43 forks source link

Update default AMI image to support G5 instances #117

Closed etiennedub closed 2 years ago

etiennedub commented 2 years ago

The current AMI image doesn't support G5 instances because the GPU driver is not included in the old cuda version (source). Updating to a more recent version of cuda fix the issue.

I tried with the AMI image Deep Learning AMI GPU CUDA 11.4.3 (Ubuntu 18.04) 20220317 and it works but I need to manually change the code.

Do you plan to reintroduce the amiId parameters or should we just update the default AMI?

etiennedub commented 2 years ago

I don't know why, but the amiId parameter works. I don't know what was wrong in my previous test.