diff --git a/build.sh b/build.sh index dc930d110f65..d921f16472d2 100755 --- a/build.sh +++ b/build.sh @@ -63,6 +63,12 @@ CUDF_JAR_JAVA_BUILD_DIR="$REPODIR/java/target" BUILD_DIRS="${LIB_BUILD_DIR} ${CUDF_BUILD_DIR} ${DASK_CUDF_BUILD_DIR} ${KAFKA_LIB_BUILD_DIR} ${CUDF_KAFKA_BUILD_DIR} ${CUSTREAMZ_BUILD_DIR} ${CUDF_JAR_JAVA_BUILD_DIR} ${PYLIBCUDF_BUILD_DIR} ${STREAMING_LIB_BUILD_DIR}" +CUDA_VERSION="${RAPIDS_CUDA_VERSION:-$(nvcc --version | sed -E -n 's/^.*release ([0-9]+\.[0-9]+).*$/\1/p')}" +if [[ -z "$CUDA_VERSION" ]]; then + echo "Could not determine CUDA version. Please set RAPIDS_CUDA_VERSION or make sure your \$PATH contains a valid nvcc." + exit 1 +fi + # Set defaults for vars modified by flags to this script VERBOSE_FLAG="" BUILD_TYPE=Release @@ -80,7 +86,7 @@ PYTHON_ARGS_FOR_INSTALL=( --no-build-isolation --no-deps --config-settings="rapidsai.disable-cuda=true" - --config-settings="rapidsai.matrix-entry=cuda=${RAPIDS_CUDA_VERSION};cuda_suffixed=false;use_cuda_wheels=false" + --config-settings="rapidsai.matrix-entry=cuda=${CUDA_VERSION};cuda_suffixed=false;use_cuda_wheels=false" ) # Set defaults for vars that may not have been defined externally