Pin to PyTorch 2.4.1 (#1796)

Lightning-AI · Oct 21, 2024 · 3ea3d93 · 3ea3d93
1 parent 5a771c7
commit 3ea3d93
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/azure-gpu-test.yml b/.github/azure-gpu-test.yml
@@ -44,8 +44,8 @@ jobs:
 
  - script: |
  pip install --upgrade pip
+ pip uninstall torch -y # uninstall so that the latest supported torch version is installed from the .toml file in the next line
  pip install '.[all,test]'
- pip install -U torch torchvision torchaudio
  displayName: 'Install dependencies'
 
  - bash: |

diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ readme = "README.md"
 license = { file = "LICENSE" }
 
 dependencies = [
- "torch>=2.2.0",
+ "torch>=2.2.0,<=2.4.1",
  "numpy<2.0",
  "lightning==2.4.0",
  "jsonargparse[signatures]>=4.30.1,<=4.32.1", # 4.33 does not seem to be compatible with Python 3.9