1.手动清空内存
查找占用显存的进程，然后kill 进程即可
sudo fuser -v /dev/nvidia* #查找占用GPU资源的PID
2. 设置显卡
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
查找占用显存的进程，然后kill 进程即可
sudo fuser -v /dev/nvidia* #查找占用GPU资源的PID
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"