diff --git a/iso/builder/build-cublas.sh b/iso/builder/build-cublas.sh index 9cf8971..386d2dd 100644 --- a/iso/builder/build-cublas.sh +++ b/iso/builder/build-cublas.sh @@ -119,6 +119,17 @@ copy_headers() { if [ -d "${from}/usr/include" ]; then cp -a "${from}/usr/include/." "${CACHE_DIR}/include/" fi + # NVIDIA CUDA packages install headers under /usr/local/cuda-X.Y/targets/x86_64-linux/include/ + find "$from" -type d -name include | while read -r inc_dir; do + case "$inc_dir" in + */usr/include) ;; # already handled above + *) + if ls "${inc_dir}"/*.h > /dev/null 2>&1; then + cp -a "${inc_dir}/." "${CACHE_DIR}/include/" + fi + ;; + esac + done } copy_libs() {