how to use the specified gpu?
Opened this issue · 2 comments
xingshuoli commented
If it is running on the server, how do I get it to run on the specified gpu?
JetRunner commented
CUDA_VISIBLE_DEVICES=1 python ...
xingshuoli commented
Thanks
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