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?