baize-chatbot icon indicating copy to clipboard operation
baize-chatbot copied to clipboard
verified publisher icon project-baize

how to use the specified gpu？

Open xingshuoli opened this issue 2 years ago • 2 comments

If it is running on the server, how do I get it to run on the specified gpu?

xingshuoli avatar Apr 24 '23 09:04 xingshuoli 

CUDA_VISIBLE_DEVICES=1 python ...

JetRunner avatar Apr 24 '23 15:04 JetRunner

Thanks

xingshuoli avatar Apr 25 '23 01:04 xingshuoli