diff --git a/aimodel/slurm-TEST-deeplabv3p-rainfall.job b/aimodel/slurm-TEST-deeplabv3p-rainfall.job
index 970396c..3083a53 100755
--- a/aimodel/slurm-TEST-deeplabv3p-rainfall.job
+++ b/aimodel/slurm-TEST-deeplabv3p-rainfall.job
@@ -16,6 +16,7 @@ module load utilities/multi
 module load readline/7.0
 module load gcc/10.2.0
 module load cuda/11.5.0
+export XLA_FLAGS="--xla_gpu_cuda_data_dir=/home/ViperAppsFiles/cuda/11.5.0"; # weird... this wasn't needed before?
 
 module load python/anaconda/4.6/miniconda/3.7
 
diff --git a/aimodel/slurm-encoderonly-rainfall.job b/aimodel/slurm-encoderonly-rainfall.job
index efac585..36f0087 100755
--- a/aimodel/slurm-encoderonly-rainfall.job
+++ b/aimodel/slurm-encoderonly-rainfall.job
@@ -16,6 +16,7 @@ module load utilities/multi
 module load readline/7.0
 module load gcc/10.2.0
 module load cuda/11.5.0
+export XLA_FLAGS="--xla_gpu_cuda_data_dir=/home/ViperAppsFiles/cuda/11.5.0"; # weird... this wasn't needed before?
 
 module load python/anaconda/4.6/miniconda/3.7