@@ -14,6 +14,7 @@ display_help() {
14
14
echo " -x | --http-proxy URL - provides URL for the environment variable http_proxy"
15
15
echo " -y | --https-proxy URL - provides URL for the environment variable https_proxy"
16
16
echo " --shared-fs-path - path to directory on shared filesystem that can be used"
17
+ echo " --skip-cuda-install - disable installing a full CUDA SDK in the host_injections prefix (e.g. in CI)"
17
18
}
18
19
19
20
function copy_build_log() {
@@ -76,6 +77,10 @@ while [[ $# -gt 0 ]]; do
76
77
export shared_fs_path=" ${2} "
77
78
shift 2
78
79
;;
80
+ --skip-cuda-install)
81
+ export skip_cuda_install=True
82
+ shift 1
83
+ ;;
79
84
-* |--* )
80
85
echo " Error: Unknown option: $1 " >&2
81
86
exit 1
@@ -195,7 +200,12 @@ ${TOPDIR}/install_scripts.sh --prefix ${EESSI_PREFIX}
195
200
# Install full CUDA SDK in host_injections
196
201
# Hardcode this for now, see if it works
197
202
# TODO: We should make a nice yaml and loop over all CUDA versions in that yaml to figure out what to install
198
- ${EESSI_PREFIX} /scripts/gpu_support/nvidia/install_cuda_host_injections.sh -c 12.1.1 --accept-cuda-eula
203
+ # Allow skipping CUDA SDK install in e.g. CI environments
204
+ if [ -z " ${skip_cuda_install} " ] || [ ! " ${skip_cuda_install} " ]; then
205
+ ${EESSI_PREFIX} /scripts/gpu_support/nvidia/install_cuda_host_injections.sh -c 12.1.1 --accept-cuda-eula
206
+ else
207
+ echo " Skipping installation of CUDA SDK in host_injections, since the --skip-cuda-install flag was passed"
208
+ fi
199
209
200
210
# Install drivers in host_injections
201
211
# TODO: this is commented out for now, because the script assumes that nvidia-smi is available and works;
0 commit comments