最近刚入门pytorch准备使用gpu训练模型,结果运行demo的时候就抛出了Runtime Error,当时我使用的是Jupyter Notebook,jupyter很贴心的给我指出引起报错的代码位置就是一句:
xxx.to(device)
这里我的device是cuda:0即gpu设备,报错因为gpu已经被其他程序使用,剩余部分不够跑我的程序,于是我结束了占用gpu的其他无用的进程
首先,在命令行输入
nvidia-smi
查看是哪些进程在占用gpu
然后我们根据描述,找到不需要的进程的PID(进程唯一标识符)
最后在命令行下结束这些不需要的进程
kill -9 PID
到此为止,大功告成,可以正常使用GPU跑模型了