RuntimeError: CUDA out of memory(已解決)

今天用pytorch訓練神經網絡時,出現如下錯誤: RuntimeError: CUDA out of memory. Tried to allocate 144.00 MiB (GPU 0; 2.00 GiB total capacity; 1.29 GiB already allocated; 79.00 MiB free; 1.30 GiB reserved in total by PyTo
相關文章
相關標籤/搜索