Update env file

torchmd · Jun 10, 2024 · 8e6ea00 · 8e6ea00
1 parent e224dc5
commit 8e6ea00
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/environment.yml b/environment.yml
@@ -6,14 +6,14 @@ dependencies:
   - matplotlib-base
   - nnpops
   - pip
-  - pytorch<2.2
-  - pytorch_geometric<2.5
-  - lightning<2.2
+  - pytorch
+  - pytorch_geometric
+  - lightning
   - pydantic
   - torchmetrics
   - tqdm
   # Dev tools
   - flake8
   - pytest
   - psutil
-  - gxx<12
+  - gxx