Skip to content

Commit 731a34f

Browse files
committed
Set cuda compute capabilities for cuda build config.
PiperOrigin-RevId: 325552145 (cherry picked from commit 9186b1a)
1 parent f57632f commit 731a34f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

.bazelrc

+1
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ build:release --copt=-msse4.2
55
# Options used to build with CUDA.
66
build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
77
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"
89

910
# Please note that MKL on MacOS or windows is still not supported.
1011
# If you would like to use a local MKL instead of downloading, please set the

0 commit comments

Comments
 (0)