RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 3.94 GiB total capacity; 3.36 GiB already allocated; 13.06 MiB free; 78.58 MiB cached)

减小batch_size, 致敬这位老哥
RuntimeError: CUDA out of memory.

相关文章: