diff --git a/aimodel/slurm-TEST-deeplabv3p-rainfall.job b/aimodel/slurm-TEST-deeplabv3p-rainfall.job
index 52bedf6..7dafb48 100755
--- a/aimodel/slurm-TEST-deeplabv3p-rainfall.job
+++ b/aimodel/slurm-TEST-deeplabv3p-rainfall.job
@@ -21,6 +21,13 @@ command_exists() {
 
 #######################################################
 
+# Fix "Could not load library libcublasLt.so.12. Error: libcublasLt.so.12: cannot open shared object file: No such file or directory" error
+if [[ "${SLURM_CLUSTER_NAME}" != "cs-cluster" ]] && [[ -d "${HOME}/cuda" ]]; then
+	echo "[slurm_runner] csgpu cluster detected, sourcing extra CUDA setup script" >&2;
+	#shellcheck source=/dev/null
+	source "${HOME}/cuda/activate.sh";
+fi
+
 # No modules on the CS cluster
 if command_exists module && [[ "${SLURM_CLUSTER_NAME}" != "cs-cluster" ]]; then
 	module load utilities/multi