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?
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?