rex-gym icon indicating copy to clipboard operation
rex-gym copied to clipboard

Published 20 hours ago • verified publisher icon nicrusso7

about training use gpu or cpu

Open zwq2018 opened this issue 4 years ago • 4 comments

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 ?

zwq2018 avatar Jun 12 '20 10:06 zwq2018

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

nicrusso7 avatar Jun 15 '20 17:06 nicrusso7

How to you start training in GPU mode?

darshisen avatar Jan 24 '21 17:01 darshisen

@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!

nicrusso7 avatar Jan 24 '21 18:01 nicrusso7

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

IverYangg avatar Feb 24 '22 12:02 IverYangg