pytorch - “加载共享库时出错：libnvinfer.so.7：无法打开共享对象文件：没有这样的文件或目录”运行 TRTorch 示例时

Question

我正在尝试使用 TRTorch 引擎编译我的 pytorch 模型。我已经根据这个链接安装了 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

Answer 1

我在 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)"

该问题可在此处找到。