RuntimeError: CUDA out of memory. Tried to allocate 192.00 MiB (GPU 0; 10.76 GiB total capacity; 9.3

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 236.00 MiB (GPU 0; 11.00_com_fang_bean的博客-CSDN博客

你可能感兴趣的:(NLP,dnn)