From 54de1065213e5a8a888203a2795eb556881fc08a Mon Sep 17 00:00:00 2001
From: rhaertel80 <robbie.haertel@gmail.com>
Date: Wed, 10 Feb 2021 14:31:49 -0700
Subject: [PATCH] Specifying compute capabilities in the TF 2.2 .bazelrc file
 to reduce startup time on GPUs; this is in line with TF 2.3 changes

---
 .bazelrc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.bazelrc b/.bazelrc
index 48bf9e9f66e..f8b6ace624d 100644
--- a/.bazelrc
+++ b/.bazelrc
@@ -6,6 +6,8 @@ build:release --copt=-msse4.2
 build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
 build:cuda --define=using_cuda=true --define=using_cuda_nvcc=true
 
+build:cuda --action_env=TF_CUDA_COMPUTE_CAPABILITIES="3.5,5.0,6.0,7.0,7.5,8.0"
+
 # Please note that MKL on MacOS or windows is still not supported.
 # If you would like to use a local MKL instead of downloading, please set the
 # environment variable "TF_MKL_ROOT" every time before build.