我正在尝试使用 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
我在 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)"
该问题可在此处获得。
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句