ElegantRL
ElegantRL copied to clipboard
Encounter cuda error running demo_A2C_PPO.py

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
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.
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!