From b7a50b24d63cdec34126648ecab6dcd71843cc6a Mon Sep 17 00:00:00 2001 From: zhoufeng Date: Tue, 23 Jun 2020 22:12:37 +0800 Subject: [PATCH] Disable cuda9.2, use cuda10.1 as default Signed-off-by: zhoufeng --- build.sh | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/build.sh b/build.sh index 608b723425..c6f80b1269 100755 --- a/build.sh +++ b/build.sh @@ -50,7 +50,7 @@ usage() echo " -D Enable dumping of function graph ir, default on" echo " -z Compile dataset & mindrecord, default on" echo " -M Enable MPI and NCCL for GPU training, gpu default on" - echo " -V Specify the minimum required cuda version, default CUDA 9.2" + echo " -V Specify the minimum required cuda version, default CUDA 10.1" echo " -I Compile predict, default off" echo " -K Compile with AKG, default off" echo " -s Enable serving module, default off" @@ -89,7 +89,7 @@ checkopts() ENABLE_DUMP_IR="on" COMPILE_MINDDATA="on" ENABLE_MPI="off" - CUDA_VERSION="9.2" + CUDA_VERSION="10.1" COMPILE_PREDICT="off" USE_GLOG="on" PREDICT_PLATFORM="" @@ -194,6 +194,10 @@ checkopts() usage exit 1 fi + if [[ "X$OPTARG" == "X9.2" ]]; then + echo "Unsupported CUDA version 9.2" + exit 1 + fi CUDA_VERSION="$OPTARG" ;; P)