Files
mlflow--mlflow/examples/pytorch/MNIST/example2/python_env.yaml
T
2026-07-13 13:22:34 +08:00

8 lines
122 B
YAML

build_dependencies:
- pip
dependencies:
- mlflow
- torch==1.8.0
- torchvision==0.9.1
- pytorch-lightning==1.0.2