module load cuda/12.3 
module load ucx/1.15.0-cuda12.3  
module load openmpi/4.1.5-cuda12.3
source /home/dp207/dp207/shared/env/production/env-base.sh 
source /home/dp207/dp207/shared/env/production/env-gpu.sh 
unset SBATCH_EXPORT