查找占用显存的进程,然后kill 进程即可
sudo fuser -v /dev/nvidia* #查找占用GPU资源的PID
import os os.environ["CUDA_VISIBLE_DEVICES"] = "0"