2021-07-14 07:51:03 +03:00
|
|
|
# Recommended because some non-commonly-used modules/examples depend on those packages.
|
|
|
|
|
|
|
|
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
|
|
|
tensorflow
|
2023-05-29 05:45:42 +03:00
|
|
|
torch == 2.0.0+cu117
|
|
|
|
|
torchvision == 0.15.0+cu117
|
|
|
|
|
pytorch-lightning >= 2.0
|
2023-01-16 08:01:51 +03:00
|
|
|
|
|
|
|
|
# for full-test-compression
|
2023-05-29 05:45:42 +03:00
|
|
|
-f https://download.openmmlab.com/mmcv/dist/cu117/torch2.0/index.html
|
2023-04-12 05:27:46 +03:00
|
|
|
mmcv >= 2.0.0rc4, < 2.1.0
|
|
|
|
|
mmdet >= 3.0
|
|
|
|
|
mmengine
|
2023-01-16 08:01:51 +03:00
|
|
|
|
2023-03-16 12:15:02 +03:00
|
|
|
git+https://github.com/microsoft/nn-Meter.git#egg=nn_meter
|
2023-08-03 13:42:09 +03:00
|
|
|
lightgbm<4.0.0
|
2021-07-14 07:51:03 +03:00
|
|
|
onnx
|
2023-03-16 12:15:02 +03:00
|
|
|
onnxsim
|
|
|
|
|
onnxruntime-gpu
|
2021-07-14 07:51:03 +03:00
|
|
|
peewee
|
|
|
|
|
graphviz
|
|
|
|
|
gym
|
2023-03-16 12:15:02 +03:00
|
|
|
sympy
|
2021-07-14 07:51:03 +03:00
|
|
|
tianshou >= 0.4.1
|
2022-06-05 07:14:40 +03:00
|
|
|
timm >= 0.5.4
|
