add matplotlib in setup #550
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# This is a basic workflow to help you get started with Actions | |
name: CI | |
on: | |
push: | |
branches: | |
- 'master' | |
- 'feature-test-siqi' | |
- 'release-v**' | |
pull_request: | |
branches: | |
- 'master' | |
# A workflow run is made up of one or more jobs that can run sequentially or in parallel | |
jobs: | |
# This workflow contains a single job called "build" | |
build: | |
# The type of runner that the job will run on | |
runs-on: ${{ matrix.os }} | |
strategy: | |
matrix: | |
python-version: [3.8] #[ 3.6, 3.7, 3.8 ] | |
torch-version: [1.7.1] #[1.7.1, 1.8.0] | |
os: [ubuntu-latest] #[ubuntu-latest, macos-latest] | |
include: | |
- os: ubuntu-latest | |
path: ~/.cache/pip | |
# - os: macos-latest | |
# path: ~/Library/Caches/pip | |
# check https://pypi.org/project/torchvision/ for torch-torchvision compatibility | |
- torch-version: 1.7.1 | |
torchvision-version: 0.8.2 | |
# - torch-version: 1.8.0 | |
# torchvision-version: 0.9.0 | |
steps: | |
- uses: actions/checkout@v2 | |
- name: Set up Python ${{ matrix.python-version }} | |
uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Cache pip | |
uses: actions/cache@v2 | |
with: | |
# This path is depend on os | |
path: ${{ matrix.path }} | |
# Look to see if there is a cache hit for the corresponding requirements file | |
key: ${{ runner.os }}-pip-py${{ matrix.python-version }}-torch==${{ matrix.torch-version }} | |
restore-keys: | | |
${{ runner.os }}-pip- | |
# - name: Install dependencies | |
# run: pip install -r requirements.txt | |
- name: Install PyTorch & Torchvision | |
run: | | |
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html | |
- name: Install other dependencies | |
run: | | |
pip install numpy | |
# - name: Test with coverage | |
# run: | | |
# pip install coverage | |
# pip install codecov | |
# coverage run --source=fedlab setup.py test | |
# - name: Upload Coverage to Codecov | |
# uses: codecov/codecov-action@v1 | |
# with: | |
# token: ${{ secrets.CODECOV_TOKEN }} | |
# name: codecov-umbrella | |
# fail_ci_if_error: true |