channels:
- conda-forge
dependencies:
- python=3.10.11
- pip<=23.1.2
- pip:
  - mlflow==2.17.0
  - cloudpickle==3.1.0
  - colorama==0.4.6
  - defusedxml==0.7.1
  - nvidia-ml-py==12.560.30
  - pandas==2.2.3
  - torch==2.5.0
  - torchvision==0.20.0
  - tqdm==4.66.5
name: mlflow-env