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.