recommended_legacy.txt 382 Bytes
Newer Older
1
-f https://download.pytorch.org/whl/torch_stable.html
2
3
torch == 1.7.1+cpu
torchvision == 0.8.2+cpu
4
5
6

# It will install pytorch-lightning 0.8.x and unit tests won't work.
# Latest version has conflict with tensorboard and tensorflow 1.x.
7
#pytorch-lightning
Yuge Zhang's avatar
Yuge Zhang committed
8
torchmetrics
9

10
11
12
onnx
peewee
graphviz
13
14
gym
tianshou >= 0.4.1
15
16
17
18
19
matplotlib

# TODO: time to drop tensorflow 1.x
keras
tensorflow < 2.0