Update environment-aurora.yml

globus-labs · Apr 10, 2024 · 4297004 · 4297004
1 parent 2505aec
commit 4297004
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/envs/environment-aurora.yml b/envs/environment-aurora.yml
@@ -38,10 +38,10 @@ dependencies:
   # PyTorch + IPEX from Intel's Conda channel
   - pytorch==2.1.0
   - intel-extension-for-pytorch==2.1.10
+  - pyg::pyg
 
   - pip
   - pip:
-    - torch_geometric
     - torch_scatter
     - pytorch-lightning<2
     - git+https://gitlab.com/ase/ase.git