Update .bazelrc
To make MediaPipe get TensorFlow’s CUDA settings copied the content from tensorflow's bazel.rc file into mediapipe's bazel.rc file
This commit is contained in:
parent
137e1cc763
commit
2eab5020a2
10
.bazelrc
10
.bazelrc
|
@ -16,6 +16,16 @@ build:linux --cxxopt=-std=c++17
|
||||||
build:linux --host_cxxopt=-std=c++17
|
build:linux --host_cxxopt=-std=c++17
|
||||||
build:linux --copt=-w
|
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
|
# windows
|
||||||
build:windows --cxxopt=/std:c++17
|
build:windows --cxxopt=/std:c++17
|
||||||
build:windows --host_cxxopt=/std:c++17
|
build:windows --host_cxxopt=/std:c++17
|
||||||
|
|
Loading…
Reference in New Issue
Block a user