recommended_gpu.txt 297 Bytes
Newer Older
qianyj's avatar
qianyj committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# Recommended because some non-commonly-used modules/examples depend on those packages.

-f https://download.pytorch.org/whl/torch_stable.html
tensorflow
torch == 1.10.0+cu113
torchvision == 0.11.1+cu113
pytorch-lightning >= 1.6.1
lightgbm
onnx
peewee
graphviz
gym
tianshou >= 0.4.1
timm >= 0.5.4