evo-design / evo

Biological foundation modeling from molecular to genome scale

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Memory usage?

bio-rat opened this issue · comments

Hi everyone,

I was trying to run the example scripts/score.py file and encountered this error of "torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 96.00 MiB. GPU"
I'm using a desktop with rtx 3070 8gb.
Do you think the GPU memory is the issue?

Yeah :/