project-baize/baize-chatbot

how to use the specified gpu?

Opened this issue · 2 comments

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

CUDA_VISIBLE_DEVICES=1 python ...

Thanks