search

openai / spinningup

An educational resource to help anyone learn deep reinforcement learning.
https://spinningup.openai.com/
MIT License
9.91k stars 2.19k forks source link

How can I use gpu with torch #302

Open ChenyangRan opened 3 years ago

ChenyangRan commented 3 years ago

Hi, I have read the doc and config. But it seems that no flag to set the use_gpu. I have test the sac_pytorch and it seems the gpu is not used. The python is 3.6 and pytorch is 1.3.1 Could you please tell how to set?

sidhantls commented 3 years ago

The implementation doesn't support GPUs. Check out https://github.com/PyTorchLightning/pytorch-lightning-bolts