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 显存释放的主要内容,如果未能解决你的问题,请参考以下文章