name: tygronai
channels:
  - pytorch
  - defaults
dependencies:
  - python=3.11
  - pip
  - pip:
    - --extra-index-url https://download.pytorch.org/whl/cu124    
    - matplotlib
    - notebook
    - pycocotools
    - torch<=2.6
    - torchvision<=0.21
    - torchaudio<=2.6
    - onnxscript==0.5.4
    - onnx==1.16
    - onnxruntime==1.18