Unverified Commit 045629c2 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Fix torch installation in github action (#462)

* bump version to 1.0.5

* fix pytorch name

* fix version

* disable the cache
parent d36850f5
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment