update torch version

allenai · Oct 9, 2024 · fce368a · fce368a
1 parent 3a3e657
commit fce368a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/actions/setup-venv/action.yml b/.github/actions/setup-venv/action.yml
@@ -11,7 +11,7 @@ inputs:
   torch-version:
     description: The PyTorch version to install
     required: false
-    default: '==2.4.0'
+    default: '==2.4.1'
 runs:
   using: composite
   steps:

diff --git a/Makefile b/Makefile
@@ -1,7 +1,7 @@
 BASE_IMAGE = ghcr.io/allenai/pytorch:2.4.1-cuda12.1-python3.11
 # NOTE: when upgrading the nightly version you also need to upgrade the torch version specification
 # in 'pyproject.toml' to include that nightly version.
-NIGHTLY_VERSION = "2.5.0.dev20240826+cu121 --index-url https://download.pytorch.org/whl/nightly/cu121"
+NIGHTLY_VERSION = "2.6.0.dev20241009+cu121 --index-url https://download.pytorch.org/whl/nightly/cu121"
 TORCHAO_VERSION = "0.5.0 --extra-index-url https://download.pytorch.org/whl/cu121"
 
 VERSION = $(shell python src/olmo_core/version.py)

diff --git a/pyproject.toml b/pyproject.toml
@@ -14,7 +14,7 @@ requires-python = ">=3.9"
 license = { file = "LICENSE" }
 dependencies = [
     "numpy",
-    "torch>=2.4,<=2.5.0.dev20240826",
+    "torch>=2.4,<=2.6.0.dev20241009",
     "cached-path",
     "requests",
     "packaging",