We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents f57632f + 731a34f commit df90ed0Copy full SHA for df90ed0
.bazelrc
@@ -5,6 +5,7 @@ build:release --copt=-msse4.2
5
# Options used to build with CUDA.
6
build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
7
build:cuda --define=using_cuda=true --define=using_cuda_nvcc=true
8
+build:cuda --action_env=TF_CUDA_COMPUTE_CAPABILITIES="sm_35,sm_37,sm_52,sm_60,sm_61,compute_70"
9
10
# Please note that MKL on MacOS or windows is still not supported.
11
# If you would like to use a local MKL instead of downloading, please set the
0 commit comments