GPU 命令行释放内存

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/liangdong2014/article/details/78860661
  • 我们在使用tensorflow+pycharm 写程序的时候, 有时候会在控制台终止掉正在运行的程序,但是有时候程序已经结束了,nvidia-smi也看到没有程序了,但是GPU的内存并没有释放,那么怎么解决该问题呢?
  • 首先执行下面的命令
sudo fuser -v /dev/nvidia* #查找占用GPU资源的PID
  • 因为我们只有一块显卡,所以显示如下图所示:
    image
  • 可以看到我们的nvidia0上还有一个python 2.7的PID
  • 然后我们执行
kill -9 pid
  • 然后再执行nvidia-smi就可以看到内存已经被释放了
展开阅读全文

没有更多推荐了,返回首页