Merge pull request #1426 from mikel-brostrom/enable-gpu-n-cpu-torch-i…

…nstallation Enable gpu n cpu torch installation
mikel-brostrom · May 11, 2024 · 97627b3 · 97627b3
2 parents 281a6be + 48912b4
commit 97627b3
Show file tree

Hide file tree

Showing 3 changed files with 490 additions and 390 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -41,7 +41,7 @@ jobs:
           python -m pip install --upgrade pip setuptools wheel
           pip install poetry
           poetry config virtualenvs.create false
-          poetry install --with yolo
+          poetry install --with cpu,yolo
 
       - name: Generate detections and embeddings
         run: |
@@ -80,7 +80,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with yolo,evolve
+          poetry install --with cpu,yolo,evolve
 
       - name: Generate detections and embeddings
         run: |
@@ -116,7 +116,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install
+          poetry install --with cpu
 
       - name: Generate detections and embeddings
         run: |
@@ -141,7 +141,7 @@ jobs:
       - run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with yolo
+          poetry install --with cpu,yolo
       - name: Test tracking with pose models
         env:
           IMG: ./assets/MOT17-mini/train/MOT17-02-FRCNN/img1/000001.jpg
@@ -160,7 +160,7 @@ jobs:
       - run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with yolo
+          poetry install --with cpu,yolo
       - name: Test tracking with pose models
         env:
           IMG: ./assets/MOT17-mini/train/MOT17-02-FRCNN/img1/000001.jpg
@@ -186,7 +186,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with yolo
+          poetry install --with cpu,yolo
       - name: Test tracking with seg models
         env:
           IMG: ./assets/MOT17-mini/train/MOT17-02-FRCNN/img1/000001.jpg
@@ -209,7 +209,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with export
+          poetry install --with cpu,export
       - name: Test export models
         run: python boxmot/appearance/reid_export.py --include torchscript onnx openvino --device cpu --batch-size 3 --dynamic
       - name: Test inference on exported models
@@ -234,7 +234,7 @@ jobs:
       - run: |
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
-          poetry install --with test
+          poetry install --with cpu,test
       - name: Test export models
         run: python boxmot/appearance/reid_export.py --include torchscript onnx openvino --device cpu --batch-size 3 --dynamic
       - name: Pytest tests  # after tracking options as this does not download models