【Ubuntu-Tensorflow】程序结束掉GPU显存没有释放的问题

来源:互联网 发布:java字符串数组 \0 编辑:程序博客网 时间:2024/06/01 09:57

笔者在ubuntu上跑Tensorflow的程序的时候,中途使用了Win+C键结束了程序的进行,但是GPU的显存却显示没有释放,一直处于被占用状态。
使用命令
watch -n 1 nvidia-smi
显示如下
这里写图片描述
两个GPU程序都在执行中,实际上GPU:0已经被笔者停止了,但是GPU没有释放,进程还在继续,所以只有采用暴力手段了,将进程手动关闭掉,进程编号如图中红线部分,由于笔者在两个GPU跑的程序一样,很难从程序名称上找到自己,却可以从GPU:num上找到自己的PID。
关闭命令如下:
sudo kill -9 PID

好,完美解决GPU显存释放问题

阅读全文
0 0
原创粉丝点击