nni/dependencies/recommended_gpu.txt

21 строка
422 B
Plaintext
Исходник Обычный вид История

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
torch == 1.10.0+cu113
torchvision == 0.11.1+cu113
pytorch-lightning >= 1.6.1
# for full-test-compression
-f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10/index.html
mmcv-full==1.7.0
mmdet
2022-03-07 10:55:40 +03:00
lightgbm
2021-07-14 07:51:03 +03:00
onnx
peewee
graphviz
gym
tianshou >= 0.4.1
timm >= 0.5.4