GPU内存溢出，以及forward() missing 1 required positional argument: ‘x‘；设置GPU

问题

1. 在实验过程里，总是遇到GPU内存溢出的问题，怎么改batchsize都不行
   表面原因设置gpu的代码有问题。究其原因，还是因为穷。
   在代码里

import os 
os.environ['CUDA_VISIBLE_DEVICES'] = "0,1,2" 

或在终端里

CUDA_VISIBLE_DEVICES=0,1

2. 清理gpu

nvidia-smi --gpu-reset -i [gpu_id]
# 或者杀掉进程
kill -9 [pid]

2. infer的时候出现

File "infer_val.py", line 168, in <module>
    _, logits = model(image, teacher=False)
TypeError: forward() missing 1 required positional argument: ‘x‘

因为空闲了三张显卡，代码自动调用了它们，然后找了半天错误，最后将gpu设置成一张就行了。os.environ[“CUDA_VISIBLE_DEVICES”] = “0”
猜测是图片无法整除的问题。。