CUDA out of memory

#8
by juliajoanna - opened

Hi,
I tried to run the same test code you provided in the model card, but I got CUDA OOM. I use A100 80GB, so it's impossible to have a better card in memory. ;) What may I do wrong?

OK, reinstalling torch helped.

juliajoanna changed discussion status to closed

Sign up or log in to comment