如何切断gpu进程

本人在对movingmnist数据集使用conlstm进行训练和测试时,会报错CUDA out of memory,当然这最主要的原因是gpu的内存不够。

但有时是jupyter notebook已经卷过一次在gpu中占了内存,但重新要卷加起来就内存不够,在这种 情况下就可以切断gpu进程,jupyter notebook中会显示kernel切断,以此重新开始。

具体操作为:

  1. cmd打开窗口。

  1. 输入nvidia-smi查看gpu型号和使用进程PID(我的为17896和19232)

  1. 以停止19232为例,输入taskkill -PID 19232 -f,终止该进程。

猜你喜欢

转载自blog.csdn.net/ArowerYoung/article/details/129016830