diff --git a/10-setup-htcondor.sh b/10-setup-htcondor.sh index 49e6a03..a240435 100755 --- a/10-setup-htcondor.sh +++ b/10-setup-htcondor.sh @@ -554,6 +554,9 @@ export condor_vars_file=$LOCAL_DIR/main/condor_vars.lst if [[ -z $OSG_DEFAULT_CONTAINER_DISTRIBUTION ]]; then OSG_DEFAULT_CONTAINER_DISTRIBUTION="10%__htc/rocky:8 90%__htc/rocky:9" fi +if [[ -z $OSG_DEFAULT_CONTAINER_DISTRIBUTION_GPU ]]; then + OSG_DEFAULT_CONTAINER_DISTRIBUTION_GPU="100%__htc/rocky:9-cuda-12.6.0" +fi # The glidein scripts expect a 1 or a 0 if is_true "$SINGULARITY_DISABLE_PID_NAMESPACES"; then SINGULARITY_DISABLE_PID_NAMESPACES=1 @@ -576,6 +579,7 @@ ERROR_GEN_PATH $PWD/error_gen.sh GLIDEIN_SINGULARITY_REQUIRE OPTIONAL GLIDEIN_Singularity_Use PREFERRED OSG_DEFAULT_CONTAINER_DISTRIBUTION $OSG_DEFAULT_CONTAINER_DISTRIBUTION +OSG_DEFAULT_CONTAINER_DISTRIBUTION_GPU $OSG_DEFAULT_CONTAINER_DISTRIBUTION_GPU SINGULARITY_IMAGE_RESTRICTIONS None SINGULARITY_DISABLE_PID_NAMESPACES $SINGULARITY_DISABLE_PID_NAMESPACES GWMS_SINGULARITY_PATH $APPTAINER_PATH