解決：RuntimeError: CUDA out of memory. Tried to allocate 2.00 MiB

有文章說：減小batch_size 解決。但是減小到4還是解決不了。   上一篇文章https://mp.csdn.net/editor/html/109211594：說with torch.no_grad()，就是不生成計算圖。計算結果沒有差異。 但是！！！對於計算機資源的佔用卻有差異！不生成計算圖，GPU就會小很大的負擔   如圖，bath_size = 4,依然報錯 在for i,(data

>>阅读原文<<