diff --git a/iso/builder/build-cublas.sh b/iso/builder/build-cublas.sh index 386d2dd..0716fac 100644 --- a/iso/builder/build-cublas.sh +++ b/iso/builder/build-cublas.sh @@ -28,6 +28,7 @@ PACKAGES_GZ="${DOWNLOAD_CACHE_DIR}/Packages.gz" echo "=== cuBLAS ${CUBLAS_VERSION} / cudart ${CUDA_USERSPACE_VERSION} / CUDA ${CUDA_SERIES} ===" if [ -f "${CACHE_DIR}/include/cublasLt.h" ] && [ -f "${CACHE_DIR}/include/cuda_runtime_api.h" ] \ + && [ -f "${CACHE_DIR}/include/crt/host_defines.h" ] \ && [ "$(find "${CACHE_DIR}/lib" \( -name 'libcublas.so*' -o -name 'libcublasLt.so*' -o -name 'libcudart.so*' \) 2>/dev/null | wc -l)" -gt 0 ]; then echo "=== cuBLAS cached, skipping download ===" echo "cache: $CACHE_DIR" @@ -155,14 +156,17 @@ CUBLAS_RT_DEB=$(download_verified_pkg "libcublas-${CUDA_SERIES_DASH}" "${CUBLAS_ CUBLAS_DEV_DEB=$(download_verified_pkg "libcublas-dev-${CUDA_SERIES_DASH}" "${CUBLAS_VERSION}") CUDART_RT_DEB=$(download_verified_pkg "cuda-cudart-${CUDA_SERIES_DASH}" "${CUDA_USERSPACE_VERSION}") CUDART_DEV_DEB=$(download_verified_pkg "cuda-cudart-dev-${CUDA_SERIES_DASH}" "${CUDA_USERSPACE_VERSION}") +CUDA_CRT_DEB=$(download_verified_pkg "cuda-crt-${CUDA_SERIES_DASH}" "${CUDA_USERSPACE_VERSION}") extract_deb "$CUBLAS_RT_DEB" "${TMP_DIR}/cublas-rt" extract_deb "$CUBLAS_DEV_DEB" "${TMP_DIR}/cublas-dev" extract_deb "$CUDART_RT_DEB" "${TMP_DIR}/cudart-rt" extract_deb "$CUDART_DEV_DEB" "${TMP_DIR}/cudart-dev" +extract_deb "$CUDA_CRT_DEB" "${TMP_DIR}/cuda-crt" copy_headers "${TMP_DIR}/cublas-dev" copy_headers "${TMP_DIR}/cudart-dev" +copy_headers "${TMP_DIR}/cuda-crt" copy_libs "${TMP_DIR}/cublas-rt" copy_libs "${TMP_DIR}/cudart-rt"