cog.yaml

build:
  gpu: true
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
    - "python3-setuptools"
  python_version: "3.10"
  cuda: "11.5" # Query nvcc --version
  python_packages:
    - "torch==2.3.1"
    - torchvision
    - torchaudio
    - "anomalib[full]==1.1.0"
    - "openvino==2024.2.0"
    - "lightning==2.3.0"
    - "scikit-learn==1.5.0"
train: "train.py:train"