forked from mindspore-Ecosystem/mindspore
Disable cuda9.2, use cuda10.1 as default
Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
This commit is contained in:
parent
890e4cbf4a
commit
b7a50b24d6
8
build.sh
8
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)
|
||||
|
|
Loading…
Reference in New Issue