PyTorch测试时候显存爆炸问题

问题描述 CUDA out of memory. Tried to allocate 2.00 MiB (GPU 0; 2.00 GiB total capacity; 1.13 GiB al ...