Не удалось найти какой-либо файл NvInferVersion.h, как мне изменить, где он его ищет?

#tensorflow #build

#тензорный поток #сборка

Вопрос:

При построении tensorflow я получаю следующую ошибку.

 Could not find any NvInferVersion.h matching version '' in any subdirectory:

    ''
    'include'
    'include/cuda'
    'include/*-linux-gnu'
    'extras/CUPTI/include'
    'include/cuda/CUPTI'
    'local/cuda/extras/CUPTI/include'
of:
    '/usr/include'
    '/usr/lib/x86_64-linux-gnu'
    '/usr/local/cuda-10.1'
 

Однако мой nvinversion.h находится по адресу /usr/include/x86_64-linux-gnu/
Как мне указать ./configure для проверки в этом месте? Где я могу добавить это в список каталогов, которые он использует?

Ответ №1:

./configure должен подсказать вам следующее:

Пожалуйста, укажите разделенный запятыми список базовых путей для поиска библиотек и заголовков CUDA. [Оставьте пустым, чтобы использовать значение по умолчанию]:

/usr/local/cuda-10.1, /lib, /lib/x86_64-linux-gnu/libfakeroot, /usr/local/cuda-10.1/цели/x86_64-linux/lib, /usr/включают/x86_64-linux-gnu/

Вы также можете убедиться, что libnvinfer установлен правильно, как описано здесь:

https://www.tensorflow.org/install/gpu