要在具体用加载模型代码的文件里面加上
import os

os.environ["CUDA_VISIBLE_DEVICES"] = "1"
才有效果