1  channels:
 2    - conda-forge
 3  dependencies:
 4    - python=3.10
 5    - pip
 6    - pip:
 7        - mlflow
 8        - torch>=2.1
 9        - torchvision>=0.15.1
10        - optuna>=3.0.0