ошибка при загрузке разделяемых библиотек: libnvinfer.so.7: невозможно открыть файл общих объектов: нет такого файла или каталога при запуске образца TRTorch

Я пытаюсь скомпилировать свою модель pytorch с движком TRTorch. Я установил TRTorch по этой ссылке.
При запуске примера кода (с помощью приведенной ниже команды из ссылки ) возникает данная ошибка:

sudo bazel run //cpp/trtorchexec -- $(realpath /home/TRTorch/tests/modules/alexnet_scripted.jit.pt) "(1,3,227,227)"

ошибка при загрузке разделяемых библиотек: libnvinfer.so.7: невозможно открыть файл общих объектов: нет такого файла или каталога

Кроме того, LD_LIBRARY_PATH установлен правильно.

экспорт LD_LIBRARY_PATH = $ LD_LIBRARY_PATH: /home/TensorRT/TensorRT-7.0.0.11/lib

Больше информации:

TRTorch: последняя версия (пакет Python и двоичный)
TensorRT: 7.0.0.11
Pytorch: 1.5.1
CUDA: 10.2
Python: 3.6


pytorch libtorch tensorrt
person Mahsa    schedule 27.07.2020    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Я задал этот вопрос в TRTorch github и исправил его с помощью:

sudo LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/TensorRT/TensorRT-7.0.0.11/lib bazel run //cpp/trtorchexec $(realpath tests/models/alexnet_traced.jit.pt) "(32 3 227 227)"

Проблема доступна здесь.

person Mahsa    schedule 01.08.2020