diff --git a/environment.yml b/environment.yml
index c28003d146cba7df7ba03864ac212c5cfc23343e..45e0ead73a7f29a047428e6e041e9d331580b616 100644
--- a/environment.yml
+++ b/environment.yml
@@ -4,11 +4,11 @@ channels:
   - rusty1s
 dependencies:
   - python
-  - pytorch
   - pip
   - pip:
     - --find-links https://data.pyg.org/whl/torch-1.9.0+cu102.html
     - scikit-learn
+    - torch
     - torch-scatter
     - torch-sparse
     - torch-cluster