"tests/vscode:/vscode.git/clone" did not exist on "81122e3d9d8a90e9e8fac6dd0952565d29ad28a0"
recommended_gpu.txt 288 Bytes
Newer Older
Yuge Zhang's avatar
Yuge Zhang committed
1
2
3
4
# Recommended because some non-commonly-used modules/examples depend on those packages.

-f https://download.pytorch.org/whl/torch_stable.html
tensorflow
5
6
7
keras == 2.4.3
torch == 1.9.0+cu111
torchvision == 0.10.0+cu111
Yuge Zhang's avatar
Yuge Zhang committed
8
pytorch-lightning >= 1.4.2
Yuge Zhang's avatar
Yuge Zhang committed
9
10
11
12
13
onnx
peewee
graphviz
gym
tianshou >= 0.4.1