torch==1.2
torchvision==0.4.0
tqdm
psutil
tensorboardX==1.8
tensorflow-gpu==1.15.2
nvidia-ml-py3
pytest
pytest-forked
pre-commit