From eb2963d9c2c57a809739caca559ce7e1b2c9bd41 Mon Sep 17 00:00:00 2001 From: Benjamin Bossan Date: Thu, 25 Jul 2024 14:51:26 +0200 Subject: [PATCH] Add PyTorch 2.4.0 to CI Also: - Remove 2.0.1 - Updgrade 2.3.0 to 2.3.1 - Use index https://download.pytorch.org/whl/torch, as torch_stable does not have 2.4.0 (yet?) --- .github/workflows/testing.yml | 7 ++----- README.rst | 4 ++-- docs/user/installation.rst | 4 ++-- 3 files changed, 6 insertions(+), 9 deletions(-) diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml index 12c2ecb4..1a01bd5c 100644 --- a/.github/workflows/testing.yml +++ b/.github/workflows/testing.yml @@ -21,11 +21,8 @@ jobs: fail-fast: false # don't cancel all jobs when one fails matrix: python_version: ['3.8', '3.9', '3.10', '3.11'] - torch_version: ['2.0.1+cpu', '2.1.2+cpu', '2.2.2+cpu', '2.3.0+cpu'] + torch_version: ['2.1.2+cpu', '2.2.2+cpu', '2.3.1+cpu', '2.4.0+cpu'] os: [ubuntu-latest] - exclude: - - python_version: '3.11' - torch_version: '2.0.1+cpu' steps: - uses: actions/checkout@v3 @@ -39,7 +36,7 @@ jobs: python -m pip install -r requirements-dev.txt python -m pip install -r requirements.txt python -m pip install pytest-pretty - python -m pip install torch==${{ matrix.torch_version }} -f https://download.pytorch.org/whl/torch_stable.html + python -m pip install torch==${{ matrix.torch_version }} -f https://download.pytorch.org/whl/torch python -m pip list - name: Install skorch run: | diff --git a/README.rst b/README.rst index 63703fcd..a68e0f0a 100644 --- a/README.rst +++ b/README.rst @@ -244,10 +244,10 @@ instructions for PyTorch, visit the `PyTorch website `__. skorch officially supports the last four minor PyTorch versions, which currently are: -- 2.0.1 - 2.1.2 - 2.2.2 -- 2.3.0 +- 2.3.1 +- 2.4.0 However, that doesn't mean that older versions don't work, just that they aren't tested. Since skorch mostly relies on the stable part of diff --git a/docs/user/installation.rst b/docs/user/installation.rst index ebc0784d..6a369339 100644 --- a/docs/user/installation.rst +++ b/docs/user/installation.rst @@ -98,10 +98,10 @@ instructions for PyTorch, visit the `PyTorch website `__. skorch officially supports the last four minor PyTorch versions, which currently are: -- 2.0.1 - 2.1.2 - 2.2.2 -- 2.3.0 +- 2.3.1 +- 2.4.0 However, that doesn't mean that older versions don't work, just that they aren't tested. Since skorch mostly relies on the stable part of