diff --git a/environment.yaml b/environment.yaml
index d8c4cd0..1d562d3 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -2,6 +2,7 @@ name: geislinger
 channels:
   - conda-forge
   - nvidia
+  - pytorch
 dependencies:
   - pip=24.2
   - python=3.11