windows釋放GPU內存方法

在進行pytorch訓練的時候,會出現類似以下錯誤:RuntimeError: CUDA out of memory. Tried to allocate 這表示顯卡內存不足,搜索怎麼釋放顯卡的內存的時候,大多都是linux下的解決方法。 windows下可以這樣做:打開cmd窗口,輸入nvidai-smi查看顯卡使用情況如圖 可以看到下邊有個PID表示進程id,然後打開任務管理器,點擊詳細信息,
相關文章
相關標籤/搜索