diff options
Diffstat (limited to 'third_party/gpus/cuda_configure.bzl')
-rw-r--r-- | third_party/gpus/cuda_configure.bzl | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/third_party/gpus/cuda_configure.bzl b/third_party/gpus/cuda_configure.bzl index 01e070f2be..a2b3e7d79e 100644 --- a/third_party/gpus/cuda_configure.bzl +++ b/third_party/gpus/cuda_configure.bzl @@ -826,8 +826,17 @@ def _cuda_autoconf_impl(repository_ctx): cuda_configure = repository_rule( implementation = _cuda_autoconf_impl, - local = True, + environ = [ + _GCC_HOST_COMPILER_PATH, + "TF_NEED_CUDA", + _CUDA_TOOLKIT_PATH, + _CUDNN_INSTALL_PATH, + _TF_CUDA_VERSION, + _TF_CUDNN_VERSION, + _TF_CUDA_COMPUTE_CAPABILITIES, + ], ) + """Detects and configures the local CUDA toolchain. Add the following to your WORKSPACE FILE: |