編碼的世界 / 優質文選 / 財富

windows釋放GPU內存方法


2022年7月07日
-   

在進行pytorch訓練的時候,會出現類似以下錯誤:RuntimeError: CUDA out of memory. Tried to allocate 這表示顯卡內存不足,搜索怎麼釋放顯卡的內存的時候,大多都是linux下的解決方法。 windows下可以這樣做:打開cmd窗口,輸入nvidai-smi查看顯卡使用情況如圖

熱門文章