linux - ImportError：没有名为“tensorrt”的模块

Question

我使用 Debian 安装在我的虚拟机上安装了 TensorRT。如果我运行“dpkg -l | grep TensorRT”，我会得到预期的结果：

ii  graphsurgeon-tf                                             5.0.2-1+cuda10.0                           amd64        GraphSurgeon for TensorRT package
ii  libnvinfer-dev                                              5.0.2-1+cuda10.0                           amd64        TensorRT development libraries and headers
ii  libnvinfer-samples                                          5.0.2-1+cuda10.0                           all          TensorRT samples and documentation
ii  libnvinfer5                                                 5.0.2-1+cuda10.0                           amd64        TensorRT runtime libraries
ii  python3-libnvinfer                                          5.0.2-1+cuda10.0                           amd64        Python 3 bindings for TensorRT
ii  python3-libnvinfer-dev                                      5.0.2-1+cuda10.0                           amd64        Python 3 development package for TensorRT
ii  tensorrt                                                    5.0.2.6-1+cuda10.0                         amd64        Meta package of TensorRT
ii  uff-converter-tf                                            5.0.2-1+cuda10.0                           amd64        UFF converter for TensorRT package

然而，当我尝试在 python 中导入它时，我得到：

File "<stdin>", line 1, in <module>
ImportError: No module named 'tensorrt'

我究竟做错了什么？

Answer 1

我有同样的错误，通过安装以下 2 个软件包解决了它：

# from TensorRT Tar Package ( TensorRT-5.0.2.6/python/ ) 
pip3 install tensorrt-5.0.2.6-py2.py3-none-any.whl  
# pycuda
pip3 install pycuda 

# libnvinfer 5.0.2-1
# libnvinfer-dev 5.0.2-1
# libnvinfer-samples 5.0.2-1
# python3-libnvinfer 5.0.2-1
# python3-libnvinfer-dev 5.0.2-1
# tensorrt 5.0.2.6

Answer 2

只需运行以下命令：

pip install nvidia-pyindex
pip install nvidia-tensorrt

Answer 3

两个建议：

1. 确保您的 TensorRT 已添加到您的路径 (LD_LIBRARY_PATH)
2. 如果您同时安装了 python 2 和 python3，请检查 pip 列表和/或 pip3 列表中的软件包安装

Answer 4

如果你是在虚拟环境中，去下面的路径看看是否存在。

(deepNN) agxxavier@agxxavier-desktop:~/miniforge3/envs/deepNN/lib/python3.6/site-packages$ ls

使用以下命令检查 tensorrt.so 所在的位置

(base) agxxavier@agxxavier-desktop:~$ sudo find / -iname "*tensorrt*"

并在您的虚拟环境中创建软连接。

(deepNN) agxxavier@agxxavier-desktop:~/miniforge3/envs/deepNN/lib/python3.6/site-packages$ ln -s /usr/lib/python3.6/dist-packages/tensorrt/tensorrt.so tensorrt.so

Answer 5

您需要在 python3 环境中导入，因为您似乎已经为 python3 安装了 TensorRT。所以它不会在 python2 上得到导入。