зеркало из https://github.com/microsoft/nni.git
24 строки
606 B
Plaintext
24 строки
606 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 == 2.0.0+cpu ; sys_platform != "darwin"
|
|
torch == 2.0.0 ; sys_platform == "darwin"
|
|
torchvision == 0.15.0+cpu ; sys_platform != "darwin"
|
|
torchvision == 0.15.0 ; sys_platform == "darwin"
|
|
pytorch-lightning >= 2.0
|
|
torchmetrics
|
|
lightgbm<4.0.0
|
|
onnx
|
|
onnxsim
|
|
onnxruntime
|
|
peewee
|
|
graphviz
|
|
gym
|
|
tianshou >= 0.4.1
|
|
matplotlib
|
|
git+https://github.com/microsoft/nn-Meter.git#egg=nn_meter
|
|
sympy
|
|
timm >= 0.5.4
|
|
typing_extensions >= 4.7.0 |