RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 748.12 MiB already allocated; 0 bytes fre
e; 796.00 MiB reserved in total by PyTorch)
My batch_ size has been set to 1. Why does this error still occur? Do I need to change to a larger GPU
Vegetebird
commented
2 years ago
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 748.12 MiB already allocated; 0 bytes fre e; 796.00 MiB reserved in total by PyTorch)
My batch_ size has been set to 1. Why does this error still occur? Do I need to change to a larger GPU