search

erwincoumans / motion_imitation

Code accompanying the paper "Learning Agile Robotic Locomotion Skills by Imitating Animals"
Apache License 2.0
1.16k stars 291 forks source link

using GPU #61

Closed code-wd closed 3 years ago

code-wd commented 3 years ago

hi, thank you for sharing you code, and I wonder to know whether this code can run with GPU. I build env for tensorflow GPU, but I find it cannot use GPU. thank you

suiup commented 3 years ago

Hi, I meet the same problem. I tried many times but I could not use the tensorflow-gpu. Looking forward to your reply. Thank you

erwincoumans commented 3 years ago

For this repo, I think we used CPU to train.