GPU 显存释放

Posted allen-rg

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了GPU 显存释放相关的知识,希望对你有一定的参考价值。

我们在使用tensorflow 的时候, 有时候会在控制台终止掉正在运行的程序,但是有时候程序已经结束了,nvidia-smi也看到没有程序了,但是GPU的内存并没有释放,那么怎么解决该问题呢?

 

首先执行下面的命令:

fuser -v /dev/nvidia* #查找占用GPU资源的PID

因为我们只有一块显卡,所以显示如下图所示: 
技术图片

可以看到我们的nvidia0上还有一个python 2.7的PID

 

然后我们执行:

kill -9 pid

然后再执行nvidia-smi就可以看到内存已经被释放了





以上是关于GPU 显存释放的主要内容,如果未能解决你的问题,请参考以下文章

GPU 显存释放

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

查询GPU时无进程运行,但是显存却被占用了

(原)tensorflow中函数执行完毕,显存不自动释放

tensorflow中使用指定的GPU及GPU显存

nvidia-smi 关键知识