Merge branch 'environment'

lus105 · lus105 · commit 294260744bf1 · 2025-12-12T15:15:07.000+02:00
diff --git a/environment.yaml b/environment.yaml
@@ -8,7 +8,7 @@ channels:
 
 dependencies:
   - python=3.11
-  - pytorch
+  - pytorch=2.5.1
   - torchvision
   - pytorch-cuda=12.4
   - lightning
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 # --------- pytorch --------- #
-torch
+torch==2.5.1
 torchvision
 lightning
 torchmetrics
