nni/dependencies/recommended_gpu.txt

15 строки
297 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
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