From 9133b1663a449dee04098d9c62dd20ffb4aec699 Mon Sep 17 00:00:00 2001
From: Geeta Chavan <geetac@google.com>
Date: Fri, 2 Oct 2020 11:07:24 -0700
Subject: [PATCH] Remove Unsupported gpu architecture compute_80
---
tensorflow/tools/ci_build/builds/libtensorflow.sh | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tensorflow/tools/ci_build/builds/libtensorflow.sh b/tensorflow/tools/ci_build/builds/libtensorflow.sh
index d0be3284eb4..a6fa334a395 100755
--- 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