From f7443db558d83351bd9c2237ceddab3c10e93293 Mon Sep 17 00:00:00 2001
From: Matteo Bunino <48362942+matbun@users.noreply.github.com>
Date: Fri, 18 Oct 2024 17:22:40 +0200
Subject: [PATCH] Update createEnvVega.sh

---
 env-files/torch/createEnvVega.sh | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/env-files/torch/createEnvVega.sh b/env-files/torch/createEnvVega.sh
index af60353e..8487d505 100644
--- 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
\ No newline at end of file
+export PIP_INDEX_TORCH_CUDA="https://download.pytorch.org/whl/cu121"
+bash env-files/torch/generic_torch.sh