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