【Ubuntu-Tensorflow】程序结束掉GPU显存没有释放的问题
生活随笔
收集整理的這篇文章主要介紹了
【Ubuntu-Tensorflow】程序结束掉GPU显存没有释放的问题
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
筆者在ubuntu上跑Tensorflow的程序的時候,中途使用了Win+C鍵結束了程序的進行,但是GPU的顯存卻顯示沒有釋放,一直處于被占用狀態。
使用命令
watch -n 1 nvidia-smi
顯示如下
兩個GPU程序都在執行中,實際上GPU:0已經被筆者停止了,但是GPU沒有釋放,進程還在繼續,所以只有采用暴力手段了,將進程手動關閉掉,進程編號如圖中紅線部分,由于筆者在兩個GPU跑的程序一樣,很難從程序名稱上找到自己,卻可以從GPU:num上找到自己的PID。
關閉命令如下:
sudo kill -9 PID
好,完美解決GPU顯存釋放問題
總結
以上是生活随笔為你收集整理的【Ubuntu-Tensorflow】程序结束掉GPU显存没有释放的问题的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 消息称字节跳动将复活“悟空问答”,改用全
- 下一篇: 【Ubuntu-Tensorflow】G