Choose GPU to use?

[yl-to]

Hi, could I add some code at the beginning of train.py to choose which GPU to use?
Like set the os.environ.

[hiram64]

As you said, the GPU to use can be set by means of CUDA_VISIBLE_DEVICES environment variable.

e.g.
(Inside script)
os.environ['CUDA_VISIBLE_DEVICES'] = "0"

or

(outside script)
export CUDA_VISIBLE_DEVICES="0"

Later, I will update the code to set GPU by tf.ConfigProto as another option.

[lzzlxxlsz]

Is cpu ok?