RuntimeError: CUDA out of memory. Tried to allocate 58.00 MiB (GPU 1; 10.76 GiB total capacity; 9.75 GiB already allocated; 25.56 MiB free; 9.89 GiB reserved in total by PyTorch)

[guangxuwang]

RuntimeError: CUDA out of memory. Tried to allocate 58.00 MiB (GPU 1; 10.76 GiB total capacity; 9.75 GiB already allocated; 25.56 MiB free; 9.89 GiB reserved in total by PyTorch)

how do you solve it？，my gpu is rtx 2080ti（memory 11G）.

[Avinash439]

I am having the same issue, Please let me know if you solve the above error?

[Avinash439]

RuntimeError: CUDA out of memory. Tried to allocate 58.00 MiB (GPU 1; 10.76 GiB total capacity; 9.75 GiB already allocated; 25.56 MiB free; 9.89 GiB reserved in total by PyTorch)

how do you solve it？，my gpu is rtx 2080ti（memory 11G）.

Just change the batch size in the file ISTR/projects/ISTR/configs/ISTR-AE-R50-3x.yaml