how to use the specified gpu？

project-baize / baize-chatbot

Let ChatGPT teach your own chatbot in hours with a single GPU!

https://arxiv.org/abs/2304.01196

GNU General Public License v3.0

3.15k stars 275 forks source link

Open xingshuoli opened 1 year ago

xingshuoli commented 1 year ago

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

JetRunner commented 1 year ago

CUDA_VISIBLE_DEVICES=1 python ...

xingshuoli commented 1 year ago

Thanks