Ubuntu跑神经网络释放显存

本文转自：Bran-Yang  https://blog.csdn.net/tintinetmilou/article/details/78947881

今天用GPU运行一个tensorflow
程序，因故意外停止后就再也不能运行程序了。

 

看警告怀疑是显存没有被释放。用nvidia-smi命令来查看一下，果不其然：

 

用命令根据PID关了对应的进程就好了

sudo kill -9 PID

本例中用sudo kill -9 2574，就可以了，再次用nvidia-smi查看，发现对应进程已经被杀死，相关显存被成功释放。