experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change RBE's default compute capability to compute_35,compute_60 (from compute_30,compute_60), because compute_30 is deprecated and will be removed in CUDA 11. We don't build releases for compute_30 and XLA requires compute_35.

Browse Source 
PiperOrigin-RevId: 314495728
Change-Id: Idf5902adb8e12073bb372592a8da6d9b2289f950
This commit is contained in:
Christian Sigg 2020-06-03 02:42:31 -07:00 committed by TensorFlower Gardener
parent 5254b55e35
commit 10c005554d
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
third_party/toolchains/preconfig/generate/generate.bzl vendored
View File

@ -41,7 +41,7 @@ def _tensorflow_rbe_config(name, compiler, python_version, os, rocm_version = No
env.update({
"TF_NEED_CUDA": "1",
"TF_CUDA_CLANG": "1" if compiler.endswith("clang") else "0",
"TF_CUDA_COMPUTE_CAPABILITIES": "3.0,6.0",
"TF_CUDA_COMPUTE_CAPABILITIES": "3.5,6.0",
"TF_ENABLE_XLA": "1",
"TF_CUDNN_VERSION": cudnn_version,
"TF_CUDA_VERSION": cuda_version,

2
third_party/toolchains/remote_config/rbe_config.bzl vendored
View File

@ -37,7 +37,7 @@ def _tensorflow_rbe_config(name, compiler, python_versions, os, rocm_version = N
env.update({
"TF_NEED_CUDA": "1",
"TF_CUDA_CLANG": "1" if compiler.endswith("clang") else "0",
"TF_CUDA_COMPUTE_CAPABILITIES": "3.0,6.0",
"TF_CUDA_COMPUTE_CAPABILITIES": "3.5,6.0",
"TF_ENABLE_XLA": "1",
"TF_CUDNN_VERSION": cudnn_version,
"TF_CUDA_VERSION": cuda_version,