ElegantRL icon indicating copy to clipboard operation
ElegantRL copied to clipboard

Encounter cuda error running demo_A2C_PPO.py

Open valleysprings opened this issue 2 years ago • 3 comments

image

valleysprings avatar Mar 04 '23 12:03 valleysprings

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

Yonv1943 avatar Mar 06 '23 02:03 Yonv1943

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 avatar Mar 06 '23 02:03 valleysprings

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!

valleysprings avatar Mar 06 '23 02:03 valleysprings