25 lines
531 B
YAML
25 lines
531 B
YAML
# 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
|