Encounter cuda error running demo_A2C_PPO.py

Question

Encounter cuda error running demo_A2C_PPO.py

Opened this issue a year ago · 3 comments

Answer 1 · 2023-03-06T02:22:25.000Z

CUDA error: out of memory

You can try args.gpu_id = -1 to use CPU instead of GPU.
Use CPU for training is much slower than GPU, but it won't meet CUDA error: out of memory

Answer 2 · 2023-03-06T02:26:11.000Z

I fix it. it's weird that after I reinstall gym and stable baseline 3, it works fine. Always meet some mysterious issues when I try using those library, very confusing.

Answer 3 · 2023-03-06T02:28:37.000Z

there are some conflicts import libs version in gym/gymnasium, SB3 and this lib. I hope you could test it in the near future. Thank you!