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