about training use gpu or cpu

nicrusso7 / rex-gym

OpenAI Gym environments for an open-source quadruped robot (SpotMicro)

Apache License 2.0

973 stars 130 forks source link

about training use gpu or cpu #5

Open zwq2018 opened 4 years ago

zwq2018 commented 4 years ago

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 commented 4 years ago

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

darshisen commented 3 years ago

How to you start training in GPU mode?

nicrusso7 commented 3 years ago

@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 commented 2 years ago

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 https://github.com/nicrusso7/rex-gym/blob/82dea26bcd8896da06240bcc3abd4de5b4696430/rex_gym/agents/scripts/configs.py#L27