Skip to content

Commit

Permalink
Add PyTorch 2.4.0 to CI (#1063)
Browse files Browse the repository at this point in the history
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?)
  • Loading branch information
BenjaminBossan authored Sep 20, 2024
1 parent e724424 commit 9ff9cfa
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 9 deletions.
7 changes: 2 additions & 5 deletions .github/workflows/testing.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -41,7 +38,7 @@ jobs:
python -m pip install -r requirements.txt
python -m pip install --force-reinstall -U "numpy<2.0.0"
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: |
Expand Down
4 changes: 2 additions & 2 deletions README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -244,10 +244,10 @@ instructions for PyTorch, visit the `PyTorch website
<http://pytorch.org/>`__. 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
Expand Down
4 changes: 2 additions & 2 deletions docs/user/installation.rst
Original file line number Diff line number Diff line change
Expand Up @@ -98,10 +98,10 @@ instructions for PyTorch, visit the `PyTorch website
<http://pytorch.org/>`__. 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
Expand Down

0 comments on commit 9ff9cfa

Please sign in to comment.