about training use gpu or cpu

[zwq2018]

hello ,i have a question for this :
if i train with gpu ,the training process will get stuck ,i do not know why,
but if i train with cpu ,all is ok.
i do not understand why ?

[nicrusso7]

Hi, I think there is an issue with GPU training. I'll test it, thanks for pointing this out.

[darshisen]

How to you start training in GPU mode?

[nicrusso7]

@darshisen: Unfortunately this feature is not tested nor properly supported. However, according with TF docs, you should be able to just change this line. Any test is appreciated!

[IverYangg]

i wonder if this flag is useful? i tried change this flag this True, but i can't see any change in speed. this line

rex-gym/rex_gym/agents/scripts/configs.py

Line 27 in 82dea26

use_gpu = False