torch==1.12.1
gym[atari]==0.21.0
opencv-python==4.8.0.76
matplotlib==3.6.2
setuptools==57.1.0
autorom[accept-rom-license]
wandb
tensorboardX
tensorboard
# pip install git+https://github.com/mila-iqia/atari-representation-learning