Update createEnvVega.sh

interTwin-eu · Nov 15, 2024 · f7443db · f7443db
1 parent e78cc0b
commit f7443db
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/env-files/torch/createEnvVega.sh b/env-files/torch/createEnvVega.sh
@@ -13,13 +13,17 @@ ml Python
 ml CMake/3.24.3-GCCcore-11.3.0
 ml mpi4py
 ml OpenMPI
-ml CUDA/11.7
+#ml CUDA/11.7
+ml CUDA/12.3
 ml GCCcore/11.3.0
-ml NCCL/2.12.12-GCCcore-11.3.0-CUDA-11.7.0
-ml cuDNN
+#ml NCCL/2.12.12-GCCcore-11.3.0-CUDA-11.7.0
+ml NCCL
+ml cuDNN/8.9.7.29-CUDA-12.3.0
+
+# You should have CUDA 12.3 now
 
 
 # Create and install torch env
 export ENV_NAME=".venv-pytorch"
-export PIP_INDEX_TORCH_CUDA="https://download.pytorch.org/whl/cu118"
-bash env-files/torch/generic_torch.sh
+export PIP_INDEX_TORCH_CUDA="https://download.pytorch.org/whl/cu121"
+bash env-files/torch/generic_torch.sh