17 lines
514 B
Plaintext
17 lines
514 B
Plaintext
# extended list of package dependencies to reach full functionality
|
|
|
|
neptune-client>=0.4.109
|
|
comet-ml>=3.1.12
|
|
mlflow>=1.0.0
|
|
test_tube>=0.7.5
|
|
wandb>=0.8.21, <0.10.0 # higher version mess with system paths
|
|
matplotlib>=3.1.1
|
|
# no need to install with [pytorch] as pytorch is already installed and torchvision is required only for Horovod examples
|
|
horovod>=0.19.2
|
|
omegaconf>=2.0.0
|
|
# scipy>=0.13.3
|
|
scikit-learn>=0.22.2
|
|
torchtext>=0.3.1, <0.7 # TODO: temporary fix fix for compatibility
|
|
onnx>=1.7.0
|
|
onnxruntime>=1.3.0
|