diff --git a/environment.yml b/environment.yml
index 870dc52..f9cacce 100644
--- a/environment.yml
+++ b/environment.yml
@@ -22,7 +22,7 @@ dependencies:
   - numpy
   - pandas<3
   - python=3.11
-  - pytorch
+  - pytorch>=2.6
   - pytorch-cuda=12.4
   - pyturbojpeg
   - scikit-learn
diff --git a/environment_CI.yml b/environment_CI.yml
index 3af1cad..edb3021 100644
--- a/environment_CI.yml
+++ b/environment_CI.yml
@@ -21,7 +21,7 @@ dependencies:
   - numpy
   - pandas<3
   - python=3.11
-  - pytorch
+  - pytorch>=2.6
   - pytest
   - pytest-asyncio>=0.23.0
   - pyturbojpeg
diff --git a/pyproject.toml b/pyproject.toml
index 6ce65ff..031d175 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -50,7 +50,7 @@ dependencies = [
     "scikit-image",
     "fitsbolt>=0.1.6",
     "toml",
-    "torch",
+    "torch>=2.6",
     "torchvision",
     "tqdm",
     "zarr>=3.0.0b0",