0
点赞
收藏
分享

微信扫一扫

安装torch2trt plugins报错解决方法:

西风白羽 2022-01-13 阅读 104
python

错误描述:RuntimeError: Error compiling objects for extension

解决方法:install with plugins · Issue #315 · NVIDIA-AI-IOT/torch2trt · GitHub

为了防止有人没办法访问github,此处放出原文内容:

```I solved it;
you must modify thesetup.py file, let it known where is you tensorRT path of include and lib

I modified it to look like this

def trt_inc_dir():
    return "/usr/local/TensorRT/include"

def trt_lib_dir():
    return "/usr/local/TensorRT/lib"

```

上述方法亲测有效。

举报

相关推荐

0 条评论