Fix torch installation in github action (#462)

* bump version to 1.0.5 * fix pytorch name * fix version * disable the cache
open-mmlab · Aug 3, 2020 · 045629c · 045629c
1 parent d36850f
commit 045629c
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -105,15 +105,15 @@ jobs:
     strategy:
       matrix:
         python-version: [3.6, 3.7]
-        torch: [1.3.1, 1.5.1]
+        torch: [1.3.1, 1.5.1+cu101]
         include:
           - torch: 1.3.1
             torchvision: 0.4.2
-          - torch: 1.5.1
-            torchvision: 0.6.1
+          - torch: 1.5.1+cu101
+            torchvision: 0.6.1+cu101
           - python-version: 3.8
-            torch: 1.5.1
-            torchvision: 0.6.1
+            torch: 1.5.1+cu101
+            torchvision: 0.6.1+cu101
 
     steps:
       - uses: actions/checkout@v2
@@ -139,7 +139,7 @@ jobs:
         run: pip install Pillow==6.2.2
         if: ${{matrix.torchvision == '0.4.2'}}
       - name: Install PyTorch
-        run: pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}}
+        run: pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} -f https://download.pytorch.org/whl/torch_stable.html
       - name: Install system dependencies
         run: sudo apt-get update && sudo apt-get install -y ffmpeg libturbojpeg
       - name: Install unittest dependencies
@@ -153,7 +153,7 @@ jobs:
           coverage report -m
       # Only upload coverage report for python3.7 && pytorch1.5
       - name: Upload coverage to Codecov
-        if: ${{matrix.torch == '1.5.1' && matrix.python-version == '3.7'}}
+        if: ${{matrix.torch == '1.5.1+cu101' && matrix.python-version == '3.7'}}
         uses: codecov/codecov-action@v1.0.10
         with:
           file: ./coverage.xml
@@ -210,7 +210,7 @@ jobs:
         run: pip install Pillow==6.2.2
         if: ${{matrix.torchvision == '0.4.2'}}
       - name: Install PyTorch
-        run: pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} -f https://download.pytorch.org/whl/torch_stable.html
+        run: pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} --no-cache-dir
       - name: Build and install
         run: |
           rm -rf .eggs