配置 gpu 可以通过 id 配置吗，目前看代码不支持通过 id，只能用 num_gpus #282

Closed

amulil started this conversation in View all discussions

amulil
Aug 28, 2023

rt.

Replies: 1 comment

Leymore
Aug 28, 2023

You can try command like CUDA_VISIBLE_DEVICES=0,1,3 python run.py blabla

Implement code is here:

https://github.com/InternLM/opencompass/blob/b2d602f42bac783bc30ef8e903177d7ab378a0de/opencompass/runners/local.py#L77-L81

0 replies

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment