Merge pull request #43742 from tensorflow/update_v

Remove Unsupported gpu architecture compute_80

Merge pull request #43742 from tensorflow/update_v
Remove Unsupported gpu architecture compute_80
5681c179 · Mihai Maruseac · GitHub · 483ccdff · 9133b166 · 5681c179
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tensorflow/tools/ci_build/builds/libtensorflow.sh tensorflow/tools/ci_build/builds/libtensorflow.sh +1 -1

未找到文件。
--- a/tensorflow/tools/ci_build/builds/libtensorflow.sh
+++ b/tensorflow/tools/ci_build/builds/libtensorflow.sh
@@ -56,7 +56,7 @@ function build_libtensorflow_tarball() {
  if [ "${TF_NEED_CUDA}" == "1" ]; then
    BAZEL_OPTS="${BAZEL_OPTS} --config=cuda --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.1:toolchain"
    export TF_NEED_ROCM=0
-    export TF_CUDA_COMPUTE_CAPABILITIES="sm_35,sm_50,sm_60,sm_70,sm_75,compute_80"
+    export TF_CUDA_COMPUTE_CAPABILITIES="sm_35,sm_50,sm_60,sm_70,sm_75"
  fi
  bazel clean --expunge
  yes "" | ./configure