From d856e0de13e2e375e40befb15d83f71378898ef4 Mon Sep 17 00:00:00 2001
From: Duc Minh La <dminh.bob.la@gmail.com>
Date: Mon, 7 Oct 2024 14:44:51 +1100
Subject: [PATCH] test

---
 job.script | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/job.script b/job.script
index 53152f3..14339e6 100644
--- a/job.script
+++ b/job.script
@@ -8,7 +8,7 @@
 #SBATCH --output=gpu_test.out
 
 module load python/3.8.5
-module load pytorch/1.7.1
+pytorch/1.7-cuda11
 module unload gcc/5.4.0
 module load cuda/11.0