-f https://download.pytorch.org/whl/torch_stable.html
tensorflow == 1.15.4
torch == 1.7.1+cpu
torchvision == 0.8.2+cpu

# It will install pytorch-lightning 0.8.x and unit tests won't work.
# Latest version has conflict with tensorboard and tensorflow 1.x.
pytorch-lightning
torchmetrics

keras == 2.1.6
onnx
peewee
graphviz
gym
tianshou >= 0.4.1
