Ubuntu pytorch 程序退出显存释放问题

在多显卡情况下，pytorch程序在退出时显存可能会没有释放。可以通过在终端运行nvidia-smi查看显存的使用。
解决方法为，在终端：

1. fusr -v /dev/nvidia* 查看在gpu上运行的所有程序
2. kill 所有连号的进程