recommended_legacy.txt 390 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

Yuge Zhang's avatar
Yuge Zhang committed
10
lightgbm
11
12
13
onnx
peewee
graphviz
14
15
gym
tianshou >= 0.4.1
16
17
18
19
20
matplotlib

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