How to set different GPU No. ?

Question

How to set different GPU No. ?

houruiyan opened this issue a year ago · 2 comments

Hi Author,

I want to run your demo code. My server have four GPU, however other colleagues also are running their own software in GPU 0. Then how can I set the number which I want to run basenji ?
Hope to get your reply. Thank you!

Best regards,
Ruiyan

Answer 1 · 2023-09-25T17:46:28.000Z

Hi Ruiyan, that’s more of a TensorFlow question than a Basenji question. Unfortunately, I don’t know how to do it.

Answer 2 · 2023-10-02T09:04:27.000Z

ok , thank you.
export CUDA_VISIBLE_DEVICES=1
This code can be used to set as an indirect method if anyone has the same problem as me.