Skip to content

Add uv environment for DeepLabCut 3.0.2 + Lightning Pose 2.4.2 - #85

Open
Hilda-Azimi wants to merge 2 commits into
neuroinformatics-unit:mainfrom
Hilda-Azimi:lp-2.4-dlc-3.0.2
Open

Hilda-Azimi wants to merge 2 commits into
neuroinformatics-unit:mainfrom
Hilda-Azimi:lp-2.4-dlc-3.0.2

Conversation

@Hilda-Azimi

Copy link
Copy Markdown

Adds examples/dlc-lightning-pose-env/: a self-contained, uv-locked Python 3.11
environment in which DeepLabCut 3.0.2 and Lightning Pose 2.4.2 install and run
together (156 pinned packages; torch 2.14.1, lightning 2.5.6, numpy 2.4.6, a
single OpenCV build). Shared as an example — it does not change poseinterface's
own dependencies.

Install: uv sync --frozen --no-install-project, then uv run python smoke_test.py.
Verified on Linux + NVIDIA T4 (both tools train, evaluate and run video inference)
and on Apple Silicon.

Notes

  • The [tool.uv] override-dependencies block resolves three conflicting OpenCV
    distributions pulled in by the two tools; opencv-contrib-python-headless is
    the single build used.
  • NVIDIA DALI (Lightning Pose's video reader) is Linux x86_64 only; LP falls back
    to OpenCV elsewhere.
  • uv lock must be run on Linux (DALI's PyPI stub must execute).
  • Pinned to Python 3.11, while poseinterface targets >=3.12. Happy to re-lock for
    3.12 if you'd prefer this integrated as a dependency group rather than an example.

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant