torch>=1.9.1
torchvision
matplotlib
numpy
gym==0.10.5
tqdm
tensorboard
pyyaml
pandas
