cog.yaml

build:
  python_version: "3.9"
  gpu: true
  cuda: "12.6"
  system_packages:
    - libgl1
    - libglx-mesa0
    - "libglib2.0-0"
    - "libgthread-2.0-0"
  python_packages:
    - h5py==3.9.0
    - matplotlib==3.7.3
    - numpy==1.24.4
    - pip==23.3.2
    - "torch==2.6.0 --extra-index-url https://download.pytorch.org/whl/cu126"
    - "torchvision==0.21.0 --extra-index-url https://download.pytorch.org/whl/cu126"
    - "torchaudio==2.6.0 --extra-index-url https://download.pytorch.org/whl/cu126"
    - json5==0.9.14
    - einops==0.7.0
    - scikit-image==0.20.0
    - timm==0.9.2
    - tqdm==4.66.1
    - wandb==0.16.1
    - opencv-python==4.8.1.78
    - imageio==2.33.1
    - mmengine==0.10.6
    - scipy==1.9.1
    - prettytable==3.10.0
    - kornia==0.7.2
    - transformers==4.36.2
    - torchmetrics==1.3.1
    - pandas==2.0.3
predict: "predict.py:Predictor"