提交 23771def 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2523 Disable cuda9.2

Merge pull request !2523 from zhoufeng/disable-cuda9-r0.5
...@@ -50,7 +50,7 @@ usage() ...@@ -50,7 +50,7 @@ usage()
echo " -D Enable dumping of function graph ir, default on" echo " -D Enable dumping of function graph ir, default on"
echo " -z Compile dataset & mindrecord, default on" echo " -z Compile dataset & mindrecord, default on"
echo " -M Enable MPI and NCCL for GPU training, gpu 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 " -I Compile predict, default off"
echo " -K Compile with AKG, default off" echo " -K Compile with AKG, default off"
echo " -s Enable serving module, default off" echo " -s Enable serving module, default off"
...@@ -88,7 +88,7 @@ checkopts() ...@@ -88,7 +88,7 @@ checkopts()
ENABLE_DUMP_IR="on" ENABLE_DUMP_IR="on"
COMPILE_MINDDATA="on" COMPILE_MINDDATA="on"
ENABLE_MPI="off" ENABLE_MPI="off"
CUDA_VERSION="9.2" CUDA_VERSION="10.1"
COMPILE_PREDICT="off" COMPILE_PREDICT="off"
USE_GLOG="on" USE_GLOG="on"
PREDICT_PLATFORM="" PREDICT_PLATFORM=""
...@@ -191,6 +191,10 @@ checkopts() ...@@ -191,6 +191,10 @@ checkopts()
usage usage
exit 1 exit 1
fi fi
if [[ "X$OPTARG" == "X9.2" ]]; then
echo "Unsupported CUDA version 9.2"
exit 1
fi
CUDA_VERSION="$OPTARG" CUDA_VERSION="$OPTARG"
;; ;;
P) P)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册