AI4Finance-Foundation / ElegantRL

Massively Parallel Deep Reinforcement Learning. 🔥
https://ai4finance.org
Other
3.62k stars 833 forks source link

Encounter cuda error running demo_A2C_PPO.py #291

Open valleysprings opened 1 year ago

valleysprings commented 1 year ago

image

Yonv1943 commented 1 year ago

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

valleysprings commented 1 year ago

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.

valleysprings commented 1 year ago

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!