зеркало из https://github.com/microsoft/nni.git
26 строки
538 B
Plaintext
26 строки
538 B
Plaintext
# Recommended because some non-commonly-used modules/examples depend on those packages.
|
|
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
tensorflow
|
|
torch == 2.0.0+cu117
|
|
torchvision == 0.15.0+cu117
|
|
pytorch-lightning >= 2.0
|
|
|
|
# for full-test-compression
|
|
-f https://download.openmmlab.com/mmcv/dist/cu117/torch2.0/index.html
|
|
mmcv >= 2.0.0rc4, < 2.1.0
|
|
mmdet >= 3.0
|
|
mmengine
|
|
|
|
git+https://github.com/microsoft/nn-Meter.git#egg=nn_meter
|
|
lightgbm<4.0.0
|
|
onnx
|
|
onnxsim
|
|
onnxruntime-gpu
|
|
peewee
|
|
graphviz
|
|
gym
|
|
sympy
|
|
tianshou >= 0.4.1
|
|
timm >= 0.5.4
|