Skip to content

Remove torch 1.11 from wheels #160

Remove torch 1.11 from wheels

Remove torch 1.11 from wheels #160

Workflow file for this run

name: Testing
on: [push, pull_request]
jobs:
pytest:
runs-on: ${{ matrix.os }}
strategy:
matrix:
os: [ubuntu-20.04]
python-version: ['3.8']
torch-version: [2.0.0, 2.1.0]
steps:
- name: Checkout with submodules
uses: actions/checkout@v3
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v4
with:
python-version: ${{ matrix.python-version }}
- name: Free up disk space
if: ${{ runner.os == 'Linux' }}
run: |
sudo rm -rf /usr/share/dotnet
- name: Setup venv
run: |
python -m venv flk-env
source flk-env/bin/activate
- name: Install setup dependencies (PyTorch ${{ matrix.torch-version }}+cpu)
run: |
source flk-env/bin/activate
echo "Python is located at $(which python)"
pip install --upgrade pip setuptools wheel ninja
pip install numpy
pip install torch==${{ matrix.torch-version }} --extra-index-url https://download.pytorch.org/whl/cpu
python -c "import torch; print('PyTorch:', torch.__version__)"
python -c "import torch; print('CUDA:', torch.version.cuda)"
python -c "import torch; print('CUDA Available:', torch.cuda.is_available())"
- name: Install main package
run: |
source flk-env/bin/activate
pip install --no-build-isolation --editable .[test]
- name: Run test-suite
run: |
source flk-env/bin/activate
pytest --cov-report=term-missing --cov-report=xml:coverage.xml --cov=falkon --cov-config pyproject.toml