DefaultCPUAllocator: can't allocate memory: you tried to allocate 3077222400 bytes. Error code 12 (Cannot allocate memory)

facebookresearch / Mask2Former

Code release for "Masked-attention Mask Transformer for Universal Image Segmentation"

MIT License

2.59k stars 388 forks source link

DefaultCPUAllocator: can't allocate memory: you tried to allocate 3077222400 bytes. Error code 12 (Cannot allocate memory) #185

Closed sushilkhadkaanon closed 1 year ago

sushilkhadkaanon commented 1 year ago

During training, it works fine. When it comes to evaluation and testing, it always show me this error? What does this mean? Have anyone faced the same issue?

sushilkhadkaanon commented 1 year ago

The issue is that It needed more GPUs. I had a single Nvidia Tesla T4. Now with 4 Tesla T4 GPUs, it is working as expected. Thanks!