search

Oneflow-Inc / one-codegeex

Apache License 2.0
7 stars 1 forks source link

如何在单机多卡上运行 #23

Open tutuxxx opened 1 year ago

tutuxxx commented 1 year ago

bash ./scripts/test_inference_oneflow.sh 0,1,2,3 ./tests/test_prompt.txt

gpu指定了4个，但实际执行只会用到第一个，导致显存不足