cog.yaml

build:
  cuda: "11.0"
  gpu: true
  python_version: "3.8"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
    - "numpy==1.21.5"
    - "ipython==7.21.0"
    - "certifi==2021.5.30"
    - "cycler==0.11.0"
    - "dominate==2.6.0"
    - "imageio==2.10.3"
    - "imageio-ffmpeg==0.4.5"
    - "kiwisolver==1.3.1"
    - "matplotlib==3.3.4"
    - "natsort==8.1.0"
    - "opencv-python==4.5.4.58"
    - "Pillow==8.4.0"
    - "protobuf==3.19.1"
    - "pyparsing==3.0.7"
    - "python-dateutil==2.8.2"
    - "PyYAML==6.0"
    - "scipy==1.5.4"
    - "six==1.16.0"
    - "sk-video==1.1.10"
    - "tensorboardX==2.4"
    - "torch==1.9.1"
    - "torchfile==0.1.0"
    - "torchvision==0.10.1"
    - "tqdm==4.62.3"
    - "typing-extensions==3.10.0.2"

predict: "predict.py:Predictor"