Add cuda compute configs to the variable list for libtensorflow GPU builds.

PiperOrigin-RevId: 330570975
Change-Id: Iafdd2711c505152cc719b2ce636a9aa18691e00f

tensorflow/tools/ci_build/linux/libtensorflow_docker.sh

@@ -58,6 +58,7 @@ ${DOCKER_BINARY} run \
   -e "TF_NEED_HDFS=0" \
   -e "TF_NEED_CUDA=${TF_NEED_CUDA}" \
   -e "TF_NEED_TENSORRT=${TF_NEED_CUDA}" \
+  -e "TF_CUDA_COMPUTE_CAPABILITIES=${TF_CUDA_COMPUTE_CAPABILITIES}" \
   -e "TF_NEED_ROCM=${TF_NEED_ROCM}" \
   -e "TF_NEED_OPENCL_SYCL=0" \
   "${DOCKER_IMAGE}" \

tensorflow/tools/ci_build/linux/libtensorflow_gpu.sh

@@ -19,4 +19,5 @@
 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"