Skip to content

Fix github action CI for GPU #783

Fix github action CI for GPU

Fix github action CI for GPU #783

# This workflow makes sure edge case versions work
name: Test Installation
on:
pull_request:
branches:
- main
paths:
- "**.py"
- ".github/workflows/**.yml"
jobs:
test-installation:
strategy:
max-parallel: 48
matrix:
os: [windows-latest, ubuntu-latest, macos-latest]
python-version: ["3.8", "3.9", "3.10"]
torch-version: [1.13.1]
runs-on: ${{ matrix.os }}
steps:
- uses: actions/checkout@v2
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v2
with:
python-version: ${{ matrix.python-version }}
- name: Install PyTorch Linux and MacOS
if: startsWith(runner.os, 'Windows') != true
run: |
./scripts/pytorch_install.sh ${{ matrix.torch-version }}
- name: Install PyTorch Windows
if: startsWith(runner.os, 'Windows')
run: |
./scripts/pytorch_install.ps1 ${{ matrix.torch-version }}
- name: Install Opacus in Editing Mode
run: |
pip install -e . --default-timeout=60
python -c "import opacus; print(opacus.__version__)"