setup.py

from setuptools import setup, find_packages

def parse_requirements(path):
    with open(path, 'r') as f:
        # strip out comments and blank lines
        return [
            line.strip()
            for line in f.splitlines()
            if line.strip() and not line.startswith('#')
        ]

# read in your two env files
cpu_reqs = parse_requirements('requirements_cpu_venv.txt')
gpu_reqs = parse_requirements('requirements_local_gpu_venv.txt')

setup(
    name="VideoAnalysisUtils",
    version="0.1.0",
    description="Utility functions and classes for video analysis",
    author="Balint Kurgyis",
    author_email="kurgyis@stanford.edu",
    packages=find_packages(),
    python_requires=">=3.8.10",

    # CPU‐only is the default install
    install_requires=cpu_reqs,

    # extras you can opt into:
    extras_require={
        # pip install .[gpu]  ⇒ CPU + GPU deps
        'gpu': gpu_reqs,
    },
)