/ examples / pytorch / MNIST / conda.yaml
conda.yaml
 1  channels:
 2    - conda-forge
 3  dependencies:
 4    - python=3.8.2
 5    - pip
 6    - pip:
 7        - mlflow
 8        - torchvision>=0.15.1
 9        - torch>=2.1
10        - lightning
11        - jsonargparse[signatures]>=4.17.0
12        - protobuf<4.0.0