Pytorch 訓練與測試時爆顯存(out of memory)的一個解決方案

Pytorch 訓練時有時候會由於加載的東西過多而爆顯存，有些時候這種狀況還能夠使用cuda的清理技術進行修整，固然若是模型實在太大，那也沒辦法。python 使用torch.cuda.empty_cache()刪除一些不須要的變量代碼示例以下：測試 try: output = model(input) except RuntimeError as exception: if "o

>>阅读原文<<