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