torch==1.10.0
numpy
tqdm
torchvision
wandb
gym==0.21.0
pybullet==3.2.0