diff --git a/devtools/conda-envs/test_env.yaml b/devtools/conda-envs/test_env.yaml
index 1cc60519..23149c67 100644
--- a/devtools/conda-envs/test_env.yaml
+++ b/devtools/conda-envs/test_env.yaml
@@ -18,7 +18,7 @@ dependencies:
     # Package dependencies
   - numpy
   - pandas
-  - pytorch
+  - pytorch<2.3
   - pydantic<2
   - lightning
   - matplotlib
diff --git a/docs/requirements.yaml b/docs/requirements.yaml
index db380cb1..6feac7e1 100644
--- a/docs/requirements.yaml
+++ b/docs/requirements.yaml
@@ -14,7 +14,7 @@ dependencies:
   # Core dependencies
   - numpy
   - pandas
-  - pytorch
+  - pytorch<2.3
   - pydantic<2 # workaround to avoid clashes with lightning
   - lightning
 
diff --git a/pyproject.toml b/pyproject.toml
index 110e7865..8dd957ce 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,7 +29,7 @@ requires-python = ">=3.8"
 # Declare any run-time dependencies that should be installed with the package.
 dependencies = [
     "lightning",
-    "torch",
+    "torch<2.3",
     "numpy",
     "pandas",
     "matplotlib",