RuntimeError: CUDA out of memory. Tried to allocate 192.00 MiB (GPU 0; 10.76 GiB total capacity; 9.36 GiB already allocated; 171.44 MiB free; 9.48 GiB reserved in total by PyTorch)
原因是epoch和batch_size太大机器受不了!
epoch全部调整为1,batch_size全部赋值调整为1。
RuntimeError: CUDA out of memory. Tried to allocate 192.00 MiB (GPU 0; 10.76 GiB total capacity; 9.36 GiB already allocated; 171.44 MiB free; 9.48 GiB reserved in total by PyTorch)
原因是epoch和batch_size太大机器受不了!
epoch全部调整为1,batch_size全部赋值调整为1。