【Pytorch】xxx.to(device) 时 Runtime Error

最近刚入门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跑模型了