nni/dependencies/recommended.txt

20 строки
494 B
Plaintext

# Recommended because some non-commonly-used modules/examples depend on those packages.
-f https://download.pytorch.org/whl/torch_stable.html
tensorflow >= 2.7.0
tensorboard >= 2.7.0
torch == 1.10.0+cpu ; sys_platform != "darwin"
torch == 1.10.0 ; sys_platform == "darwin"
torchvision == 0.11.1+cpu ; sys_platform != "darwin"
torchvision == 0.11.1 ; sys_platform == "darwin"
pytorch-lightning >= 1.6.1
torchmetrics
lightgbm
onnx
peewee
graphviz
gym
tianshou >= 0.4.1
matplotlib
timm >= 0.5.4