Not able to run the TRPO example on GPU

Khrylx / PyTorch-RL

PyTorch implementation of Deep Reinforcement Learning: Policy Gradient methods (TRPO, PPO, A2C) and Generative Adversarial Imitation Learning (GAIL). Fast Fisher vector product TRPO.

MIT License

1.09k stars 186 forks source link

Not able to run the TRPO example on GPU #12

Closed avijit9 closed 5 years ago

avijit9 commented 5 years ago

How to run the TRPO example on GPU? I am getting an error at this: https://github.com/Khrylx/PyTorch-RL/blob/master/core/trpo.py#L83

Khrylx commented 5 years ago

What error did you get? Is it this one https://github.com/Khrylx/PyTorch-RL/issues/1#issue-282815581? One solution would be upgrade your CUDA version to 9.0.