recommended.txt 470 Bytes
Newer Older
1
2
3
# Recommended because some non-commonly-used modules/examples depend on those packages.

-f https://download.pytorch.org/whl/torch_stable.html
4
5
6
7
8
9
10
tensorflow == 2.5.0
keras == 2.4.3
tensorboard == 2.5.0
torch == 1.9.0+cpu ; sys_platform != "darwin"
torch == 1.9.0 ; sys_platform == "darwin"
torchvision == 0.10.0+cpu ; sys_platform != "darwin"
torchvision == 0.10.0 ; sys_platform == "darwin"
Yuge Zhang's avatar
Yuge Zhang committed
11
12
pytorch-lightning >= 1.5
torchmetrics
13
14
15
onnx
peewee
graphviz
16
17
gym
tianshou >= 0.4.1