Update cuda runtime environment to 11.8

globus-labs · May 20, 2024 · 96cd157 · 96cd157
1 parent b85385b
commit 96cd157
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/envs/environment-cuda117.yml → envs/environment-cuda11.yml b/envs/environment-cuda117.yml → envs/environment-cuda11.yml
@@ -18,7 +18,7 @@ dependencies:
 
   # Tools used as part of the ML packages
   - imageio
-  - pytorch-lightning<2
+  - pytorch-lightning
   - wandb
 
   # Tools to work with crystalline and molecular data
@@ -30,9 +30,9 @@ dependencies:
   - conda-forge::raspa2
 
   # Use Conda PyTorch to avoid OpenMP disagreement with other libraries
-  - pytorch::pytorch==1.13.1
-  - pytorch::pytorch-cuda=11.7
-  - nvidia::cudatoolkit=11.7
+  - pytorch::pytorch==2.3.*
+  - pytorch::pytorch-cuda=11.8
+  - cuda-version==11.8
 
   # Services used by the workflow
   - redis==5.*