# environment.yml
name: storyboard-video
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - python=3.11
  - pip=24.3.1

  # FFmpeg in env is easiest on Windows if you rely on conda-forge
  - ffmpeg=7.1

  # PyTorch with CUDA runtime bundled (NOT using local CUDA 13.1 toolkit)
  # Using compatible versions for PyTorch 2.5.1 with CUDA 12.4
  - pytorch=2.5.1
  - pytorch-cuda=12.4
  - torchvision=0.20.1
  - torchaudio=2.5.1

  # pip deps (mirrors requirements.txt)
  - pip:
      - -r requirements.txt