Not able to run the TRPO example on GPU
avijit9 opened this issue · 1 comments
avijit9 commented
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
What error did you get? Is it this one #1 (comment)?
One solution would be upgrade your CUDA version to 9.0.