Skip to content

Commit 3a9b697

Browse files
authored
Update .bazelrc
Fix #1808 to specific correct capabilities to match up 2.3 release of serving. Notably remove compute_80 as it only is available in cuda 11, and 2.2/3 releases use cuda 10.
1 parent cfe83b1 commit 3a9b697

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

.bazelrc

+1-2
Original file line numberDiff line numberDiff line change
@@ -5,8 +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-
9-
build:cuda --action_env=TF_CUDA_COMPUTE_CAPABILITIES="3.5,5.0,6.0,7.0,7.5,8.0"
8+
build:cuda --action_env=TF_CUDA_COMPUTE_CAPABILITIES="3.5,3.7,5.2,6.0,6.1,7.0"
109

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

0 commit comments

Comments
 (0)