唉 显卡显存爆了。。。 解决方法就是禁用显卡跑 使用CPU ~ import os os.environ["CUDA_VISIBLE_DEVICES"]="-1" 这个方法也可以使用于只想用CPU处理模型的时候。