diff --git a/coffea-basev0/environment.yaml b/coffea-basev0/environment.yaml
index d470676..f6ae182 100644
--- a/coffea-basev0/environment.yaml
+++ b/coffea-basev0/environment.yaml
@@ -40,7 +40,7 @@ dependencies:
   - hist
     # ML
   - xgboost
-  - pytorch=1.12
+  - pytorch
   - torch-scatter
   - pip
   - coffea=0.7.22
diff --git a/coffea-dask/environment.yaml b/coffea-dask/environment.yaml
index 67b8ea6..4329707 100644
--- a/coffea-dask/environment.yaml
+++ b/coffea-dask/environment.yaml
@@ -40,7 +40,7 @@ dependencies:
   - hist
     # ML
   - xgboost
-  - pytorch=1.12
+  - pytorch
   - torch-scatter
   - pip
   - coffea=2024.1.2