用只有2个G的显卡跑数据就需要在训练之前先把无关进程杀掉,防止跑到一半显存满了
nvidia-smi:显示当前GPU中的线程
nvidia-smi
kill -9 PID:输入PID以结束线程
kill -9 PID