提交 209e2175 编写于 作者: C caisq

Merge pull request #2684 from caisq/r0.9-ver

parameterized_docker_build includes three CUDA capabilities
...@@ -179,6 +179,9 @@ if [[ "${DO_PIP_BUILD}" == "1" ]]; then ...@@ -179,6 +179,9 @@ if [[ "${DO_PIP_BUILD}" == "1" ]]; then
export TF_BUILD_IS_OPT="OPT" export TF_BUILD_IS_OPT="OPT"
export TF_BUILD_IS_PIP="PIP" export TF_BUILD_IS_PIP="PIP"
export TF_BUILD_APPEND_CI_DOCKER_EXTRA_PARAMS=\
"-e TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2"
pushd "${SCRIPT_DIR}/../../../" pushd "${SCRIPT_DIR}/../../../"
rm -rf pip_test/whl && rm -rf pip_test/whl &&
tensorflow/tools/ci_build/ci_parameterized_build.sh tensorflow/tools/ci_build/ci_parameterized_build.sh
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册