Сборка Tensorflow 1.4.1 и ссылка на libcufft_static.a - PullRequest
0 голосов
/ 06 сентября 2018

Я работаю над большим проектом, в котором Cuda v8.0 используется во многих отдельно создаваемых библиотеках. В одном из них cufft должен быть связан статически, так как мы используем обратные вызовы cufft: https://devblogs.nvidia.com/cuda-pro-tip-use-cufft-callbacks-custom-data-processing/

Другая библиотека проекта должна использовать тензор потока c-api. К сожалению, libtensorflow.so обычно связывается с общей библиотекой libcufft.so, что приводит к сбою программы при загрузке библиотек с обратными вызовами cufft и тензорным потоком.

Лучшее решение, которое я придумала на данный момент, - это самостоятельно построить tenorflow и связать его со статической версией cufft. Как мне нужно адаптировать команду / файлы bazel build в tenorflow для использования libcufft_static.a?

Или кто-то может предложить мне лучшее решение, чтобы исправить повреждение, вызванное загрузкой libcufft.so и библиотекой обратного вызова, которая статически связана с cufft?

...