[Bug] 无法指定gpu

[BruceCheng2025]

描述该错误

无法指定gpu，无论在shell中使用CUDA_VISIBLE_DEVICES=1，还是在python中使用os.environ["CUDA_VISIBLE_DEVICES"]=1，都无法让程序只占用1卡，程序永远都会优先占用0卡。

环境信息

shell中

CUDA_VISIBLE_DEVICES=1 python run.py configs/eval_llama_7b.py -w outputs/llama_7b --gpu_idx=1

python中

parser = argparse.ArgumentParser(description='Run an evaluation task')
parser.add_argument('--gpu_idx', type=str, default=None)
args = parser.parse_args()
os.environ["CUDA_VISIBLE_DEVICES"] = args.gpu_idx

其他信息

No response

[BruceCheng2025]

将opencompass/opencompass/runners/local.py文件(A位置)中的B区域代码注释，并添加A代码，即可。

[tonysy]

PR is wlecomed for improving OpenCompass. Thanks in advance.