Less VRAM U_U 14GBvRAM in USE

FangShancheng / ABINet

Read Like Humans: Autonomous, Bidirectional and Iterative Language Modeling for Scene Text Recognition

Other

420 stars 72 forks source link

Less VRAM U_U 14GBvRAM in USE #76

Open sneedgers opened 2 years ago

sneedgers commented 2 years ago

hello my friend. From Germany!!!!

can you help me to use less vram? lol ?

RuntimeError: CUDA out of memory. Tried to allocate 768.00 MiB (GPU 0; 14.76 GiB total capacity; 12.85 GiB already allocated; 435.75 MiB free; 13.22 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

sneedgers commented 2 years ago

Fixed

I used a less agressive config