Add torch==1.7.0 to CI tests (#8916)

* Add torch==1.7.0 to CI tests * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Attempt crash on 1.7.0 * revert crash * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml * Update ci-testing.yml
ultralytics · Aug 9, 2022 · e242ede · e242ede
1 parent 7ab1f50
commit e242ede
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
@@ -60,6 +60,10 @@ jobs:
           - os: ubuntu-latest
             python-version: '3.9'
             model: yolov5n
+          - os: ubuntu-latest
+            python-version: '3.8'  # torch 1.7.0 requires python >=3.6, <=3.8
+            model: yolov5n
+            torch: '1.7.0'  # min torch version CI https://pypi.org/project/torchvision/
     steps:
       - uses: actions/checkout@v3
       - uses: actions/setup-python@v4
@@ -78,10 +82,12 @@ jobs:
       - name: Install requirements
         run: |
           python -m pip install --upgrade pip wheel
-          pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu
-          python --version
-          pip --version
-          pip list
+          if [ "${{ matrix.torch }}" == "1.7.0" ]; then
+              pip install -r requirements.txt torch==1.7.0 torchvision==0.8.1 --extra-index-url https://download.pytorch.org/whl/cpu
+          else
+              pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu
+          fi
+        shell: bash  # required for Windows compatibility
       - name: Check environment
         run: |
           python -c "import utils; utils.notebook_init()"
@@ -91,6 +97,9 @@ jobs:
           echo "GITHUB_ACTOR is ${{ github.actor }}"
           echo "GITHUB_REPOSITORY is ${{ github.repository }}"
           echo "GITHUB_REPOSITORY_OWNER is ${{ github.repository_owner }}"
+          python --version
+          pip --version
+          pip list
       - name: Run tests
         shell: bash
         run: |