hiyouga / LLaMA-Factory

Unify Efficient Fine-Tuning of 100+ LLMs
Apache License 2.0
25.52k stars 3.16k forks source link

Yi-1.5-9B推理gpu利用率为0 #4628

Closed Jack-mi closed 1 day ago

Jack-mi commented 2 days ago

Reminder

System Info

【llamafactory-cli】CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 llamafactory-cli train config_infer.yaml
【config_infer.yaml】

image

Reproduction

【gpu利用率】

image

Expected behavior

No response

Others

No response

Jack-mi commented 2 days ago

只有0号卡在跑,其余利用率都为0

hiyouga commented 1 day ago

建议更新代码。 image

Jack-mi commented 12 hours ago
image

我已经pull了最新的llamafactory,但还是利用率不平衡。或者是我的infer脚本有啥问题?你这也不说