requirements.txt

wheel==0.38.4
setuptools==65.5.0
torch>=2.2.2
torchvision>=0.17.2
cython<3
dm_control==1.0.11
einops==0.6.0
gdown==4.7.1
gym==0.25.2
hydra-core==1.3.1
mujoco-py<2.2,>=2.1
numpy==1.23.5
pytorch_lightning==2.0.0
scikit-learn
shimmy==0.2.0
wandb==0.14.0
termcolor
decord==0.6.0
gradio>=4.0.0
allenact@git+https://github.com/allenai/allenact.git@usd#egg=allenact&subdirectory=allenact