CUDA doesn't work on latest Amazon Deep Learning AMI

saharmor / dalle-playground

A playground to generate images from any text prompt using Stable Diffusion (past: using DALL-E Mini)

MIT License

2.77k stars 596 forks source link

CUDA doesn't work on latest Amazon Deep Learning AMI #73

Open macsj200 opened 2 years ago

macsj200 commented 2 years ago

The latest DLAMI based on Ubuntu doesn't appear to work with this project:

https://aws.amazon.com/marketplace/pp/prodview-x5nivojpquy6y

I end up with a bunch of errors reading something like: upgrade CuDNN, no fallback algorithm found, etc.

Is there a way to determine which version of CUDA/CuDNN will work with this repo?

davisengeler commented 2 years ago

It’s working for me with cuDNN 8.4.1 for CUDA 11.7 on Ubuntu 22.04. Not sure if this is helpful for you, but there’s a cuDNN Support Matrix which shows version compatibilities. Out of curiosity, which model are you running? (mini, Mega, or Mega_full).