Merge pull request #1478 from mikel-brostrom/cpu-torch-ci-installation

cpu torch install in CI by simply sed toml
mikel-brostrom · Jun 13, 2024 · 9423fae · 9423fae
2 parents 34e3060 + eff4e1b
commit 9423fae
Showing 1 changed file with 65 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -38,9 +38,16 @@ jobs:
       - name: Install requirements
         shell: bash  # for Windows compatibility
         run: |
-          python -m pip install --upgrade pip setuptools wheel
-          pip install poetry
+          if [[ "$OSTYPE" == "darwin"* ]]; then
+            # macOS
+            sed -i '' 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
+            # Linux
+            sed -i 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          fi
+          python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
+          poetry lock --no-update
           poetry install --with yolo
 
       - name: Generate detections and embeddings
@@ -78,8 +85,16 @@ jobs:
       - name: Install requirements
         shell: bash  # for Windows compatibility
         run: |
+          if [[ "$OSTYPE" == "darwin"* ]]; then
+            # macOS
+            sed -i '' 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
+            # Linux
+            sed -i 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          fi
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
+          poetry lock --no-update
           poetry install --with yolo,evolve
 
       - name: Generate detections and embeddings
@@ -114,8 +129,16 @@ jobs:
       - name: Install requirements
         shell: bash  # for Windows compatibility
         run: |
+          if [[ "$OSTYPE" == "darwin"* ]]; then
+            # macOS
+            sed -i '' 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
+            # Linux
+            sed -i 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          fi
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
+          poetry lock --no-update
           poetry install
 
       - name: Generate detections and embeddings
@@ -139,8 +162,16 @@ jobs:
         with:
           python-version: '3.11'
       - run: |
+          if [[ "$OSTYPE" == "darwin"* ]]; then
+            # macOS
+            sed -i '' 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
+            # Linux
+            sed -i 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          fi
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
+          poetry lock --no-update
           poetry install --with yolo
       - name: Test tracking with pose models
         env:
@@ -158,8 +189,16 @@ jobs:
         with:
           python-version: '3.11'
       - run: |
+          if [[ "$OSTYPE" == "darwin"* ]]; then
+            # macOS
+            sed -i '' 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
+            # Linux
+            sed -i 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          fi
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
+          poetry lock --no-update
           poetry install --with yolo
       - name: Test tracking with pose models
         env:
@@ -184,8 +223,16 @@ jobs:
           python-version: '3.11'
       - name: Install dependencies
         run: |
+          if [[ "$OSTYPE" == "darwin"* ]]; then
+            # macOS
+            sed -i '' 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
+            # Linux
+            sed -i 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          fi
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
+          poetry lock --no-update
           poetry install --with yolo
       - name: Test tracking with seg models
         env:
@@ -207,8 +254,16 @@ jobs:
           python-version: '3.11'
       - name: Install dependencies
         run: |
+          if [[ "$OSTYPE" == "darwin"* ]]; then
+            # macOS
+            sed -i '' 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
+            # Linux
+            sed -i 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          fi
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
+          poetry lock --no-update
           poetry install --with export
       - name: Test export models
         run: python boxmot/appearance/reid_export.py --include torchscript onnx openvino --device cpu --batch-size 3 --dynamic
@@ -232,8 +287,16 @@ jobs:
         with:
           python-version: '3.11'
       - run: |
+          if [[ "$OSTYPE" == "darwin"* ]]; then
+            # macOS
+            sed -i '' 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
+            # Linux
+            sed -i 's/source="torch_cuda121"/source="torchcpu"/g' pyproject.toml
+          fi
           python -m pip install --upgrade pip setuptools wheel poetry
           poetry config virtualenvs.create false
+          poetry lock --no-update
           poetry install --with test
       - name: Test export models
         run: python boxmot/appearance/reid_export.py --include torchscript onnx openvino --device cpu --batch-size 3 --dynamic