Merge pull request #43713 from geetachavan1/cherrypicks_XBAAK

[Cherrypick:r2.3] Fix libtensorflow CUDA compute capabilities.

tensorflow/tools/ci_build/builds/libtensorflow.sh

@@ -56,6 +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"
   fi
   bazel clean --expunge
   yes "" | ./configure

tensorflow/tools/ci_build/linux/libtensorflow_gpu.sh

@@ -19,5 +19,4 @@
 set -ex
 SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 export TF_NEED_CUDA=1
-export TF_CUDA_COMPUTE_CAPABILITIES="sm_35,sm_50,sm_60,sm_70,sm_75,compute_80"
 "${SCRIPT_DIR}/libtensorflow_docker.sh"