torch>=1.2
torchvision>=0.4.0
pillow==6.2.2
tqdm
psutil
tensorboardX==1.8
tensorflow-gpu==1.15.2
pytest
pytest-forked
pre-commit
