search

esteveste / dreamerV2-pytorch

Pytorch implementation of DreamerV2: Mastering Atari with Discrete World Models, based on the original implementation
17 stars 4 forks source link

Version of gym library #4

Open realjoenguyen opened 1 year ago

realjoenguyen commented 1 year ago

Hi, I can't run

python3 dreamerv2/train.py --logdir ~/logdir/atari_pong/dreamerv2/1 --configs defaults atari --task atari_pong

after installing packages in requirements.txt. Here is my pip freeze

gym==0.26.2 gym-notices==0.0.8 atari-py==0.2.6 pygame==1.9.6

May I ask the exact versions you used for gym and gym[atari]? Thank you

Updated After installing all versions of gym and atari, I got this error AttributeError: 'numpy.random._generator.Generator' object has no attribute 'randint'

So may I ask the exact versions of gym, numpy, gym[atari] you used? Thank you!

esteveste commented 1 year ago

Hello, I think this also happened to me due to a dependency problem with newer version of numpy or gym.

This is the versions I used for this project: numpy 1.20.3
gym 0.21.0 ale-py 0.7.3 atari-py 0.2.9

Hope it helps!