你好,我在epoch0加载到67%时报错内存不足,可以给出点见解嘛,我使用的是两张2080ti,batch_size=1,TRAIN_IMG_SIZE=640.
RuntimeError: CUDA out of memory. Tried to allocate 314.00 MiB (GPU 1; 10.76 GiB total capacity; 8.53 GiB already allocated; 179.44 MiB free; 9.46 GiB reserved in total by PyTorch)
你好,我在epoch0加载到67%时报错内存不足,可以给出点见解嘛,我使用的是两张2080ti,batch_size=1,TRAIN_IMG_SIZE=640.
RuntimeError: CUDA out of memory. Tried to allocate 314.00 MiB (GPU 1; 10.76 GiB total capacity; 8.53 GiB already allocated; 179.44 MiB free; 9.46 GiB reserved in total by PyTorch)