Как передать флаги nv cc при построении тензорного потока с базелем? - PullRequest
0 голосов
/ 26 апреля 2020

Я хочу передать некоторые пользовательские флаги nv cc во время сборки, например, сказать nv cc, чтобы динамически связать библиотеку времени выполнения cuda с флагом --cudart = shared. На windows 10 в Visual Studio 2015 я мог бы просто использовать следующую командную строку:

bazel build --config=opt --config=cuda --copt=-nvcc_options=cudart=shared //tensorflow/tools/pip_package:build_pip_package

и msv c просто передаст флаг, указанный в -nvcc_options, nv cc, но на Linux он не работает, появляются следующие ошибки:

ERROR: /root/.cache/bazel/_bazel_root/5ac94cf1f70fd575be3807b8a4a32ede/external/com_google_absl/absl/debugging/BUILD.bazel:190:1: C++ compilation of rule '@com_google_absl//absl/debugging:leak_check' failed (Exit 1)
gcc: error: unrecognized command line option '-nvcc_options=cudart=shared'
Target //tensorflow/tools/pip_package:build_pip_package failed to build
Use --verbose_failures to see the command lines of failed build steps.
INFO: Elapsed time: 152.342s, Critical Path: 48.24s
INFO: 1236 processes: 1236 local.
FAILED: Build did NOT complete successfully

Платформа и распространение ОС (например, Linux Ubuntu 16.04): Ubuntu 18.04 Mobile device (например, iPhone 8, Pixel 2, Samsung Galaxy), если проблема возникает на мобильном устройстве: TensorFlow установлен из (источника или двоичного файла): источник Версия TensorFlow: r1.14 Python версия: 3.6.9 Установлен с помощью virtualenv? Пип? conda ?: сборка в сырой среде linux, с установленным pip. Версия Bazel (при компиляции из источника): 0.24.1 GCC / версия компилятора (при компиляции из источника): 4.8.5 CUDA / cuDNN версия: CUDA 10.0 CUDNN 7.6. Модель с 5 графическими процессорами и памятью: Nvidia GTX 1660Ti 6 ГБ

ERROR: /root/.cache/bazel/_bazel_root/5ac94cf1f70fd575be3807b8a4a32ede/external/com_google_absl/absl/debugging/BUILD.bazel:190:1: C++ compilation of rule '@com_google_absl//absl/debugging:leak_check' failed (Exit 1)
gcc: error: unrecognized command line option '-nvcc_options=cudart=shared'
Target //tensorflow/tools/pip_package:build_pip_package failed to build
Use --verbose_failures to see the command lines of failed build steps.
INFO: Elapsed time: 152.342s, Critical Path: 48.24s
INFO: 1236 processes: 1236 local.
FAILED: Build did NOT complete successfully
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...