写在前面
我安装了tensorRT,在运行它自带的模型都没问题。
但是在代码中import tensorrt就报错:
ModuleNotFoundError: No module named ‘tensorrt’。
网上搜了一大堆,发现是没有在自己的python环境下安装。
所以特意写这篇文章记录一下。
在进行下一步骤之前,要确保自己安装了tensorRT并添加了环境变量。
安装过程我就不多说了,自己找文章吧。
问题解决
解决ModuleNotFoundError: No module named ‘tensorrt’问题:
cd TensorRT-${version}/python
python3 -m pip install tensorrt-*-cp3x-none-linux_x86_64.whl
解释下这两句命令。
第一句是进入tensorRT的安装目录TensorRT-x.x.x,在此目录下有个文件夹python,进入这个文件夹:
在python文件夹下,输入ls,查看目录内容,比如我的如下:
在此文件夹下,根据你的python版本找到whl文件,如我的python是3.8,我找的就是tensorrt-8.6.1-cp38-none-linux_x86_64.whl文件。
输入命令行:
python3 -m pip install tensorrt-8.6.1-cp38-none-linux_x86_64.whl
即可下载。
测试
在python解释器下,输入import tensorrt,如果没有报错,即是正确安装。
问题解决!