From 2eab5020a28b63dcd0072340d338ec85a647054c Mon Sep 17 00:00:00 2001 From: aishwarya123mathpati <57386775+aishwarya123mathpati@users.noreply.github.com> Date: Wed, 22 Sep 2021 10:25:56 +0530 Subject: [PATCH] Update .bazelrc MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit To make MediaPipe get TensorFlow’s CUDA settings copied the content from tensorflow's bazel.rc file into mediapipe's bazel.rc file --- .bazelrc | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/.bazelrc b/.bazelrc index 37a0bc114..6fa9f5f0e 100644 --- a/.bazelrc +++ b/.bazelrc @@ -16,6 +16,16 @@ build:linux --cxxopt=-std=c++17 build:linux --host_cxxopt=-std=c++17 build:linux --copt=-w +# This config refers to building with CUDA available. It does not necessarily +# mean that we build CUDA op kernels. +build:using_cuda --define=using_cuda=true +build:using_cuda --action_env TF_NEED_CUDA=1 +build:using_cuda --crosstool_top=@local_config_cuda//crosstool:toolchain + +# This config refers to building CUDA op kernels with nvcc. +build:cuda --config=using_cuda +build:cuda --define=using_cuda_nvcc=true + # windows build:windows --cxxopt=/std:c++17 build:windows --host_cxxopt=/std:c++17