From 5015d1e01b60894935327a4450d86d78ff0f2a99 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 Dec 2024 15:48:02 +0100 Subject: [PATCH] `uv pip install` for Benchmarks (#17749) Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com> Co-authored-by: Laughing-q <1185102784@qq.com> --- .github/workflows/ci.yaml | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index 7c504d94..e0eff35c 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -102,21 +102,19 @@ jobs: python-version: ["3.11"] model: [yolo11n] steps: + - uses: astral-sh/setup-uv@v3 - uses: actions/checkout@v4 - uses: actions/setup-python@v5 with: python-version: ${{ matrix.python-version }} - cache: "pip" # caching pip dependencies - name: Install requirements shell: bash # for Windows compatibility run: | - # Warnings: uv causes numpy errors during benchmarking - python -m pip install --upgrade pip wheel - pip install -e ".[export]" "coverage[toml]" --extra-index-url https://download.pytorch.org/whl/cpu + uv pip install --system -e ".[export]" "coverage[toml]" --extra-index-url https://download.pytorch.org/whl/cpu --index-strategy unsafe-first-match - name: Check environment run: | yolo checks - pip list + uv pip list - name: Benchmark DetectionModel shell: bash run: coverage run -a --source=ultralytics -m ultralytics.cfg.__init__ benchmark model='path with spaces/${{ matrix.model }}.pt' imgsz=160 verbose=0.309