search

hujiecpp / ISTR

ISTR: End-to-End Instance Segmentation with Transformers (https://arxiv.org/abs/2105.00637)
202 stars 28 forks source link

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) #21

Open guangxuwang opened 2 years ago

guangxuwang commented 2 years ago

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 commented 11 months ago

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

Avinash439 commented 11 months ago

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