CUDA error help

[Wenzhe-Ma]

I encountered this error with the e2e script, with my protein of 255AA in length. The error happens at the last step when building the 3D models. Any idea why this is happening and what can I do to fix it? I am using a GTX1660, a better setup is not possible in the near future.

Thanks,
Wenzhe

RuntimeError: CUDA out of memory. Tried to allocate 456.00 MiB (GPU 0; 5.80 GiB
total capacity; 3.24 GiB already allocated; 420.88 MiB free; 3.66 GiB reserved i
n total by PyTorch)

[jong3]

Hi, I also meet the same problem. Have you solved this problem now ？

[Wenzhe-Ma]

No I have not. Sorry. I don't have experience on GPU programming and debugging, and have no idea what the problem was.