描述 当我尝试在 python 虚拟环境中使用 pip 安装tensorrt时,安装失败并给出以下错误: 错误:张量构建轮失败。
尝试过可能的解决方案 我已经升级了pip的版本了,还是不行。
环境 TensorRT 版本:8.5 CUDA版本:11.4 CUDNN 版本:8.6 操作系统+版本:Jetson 35.2 Python 版本(如果适用):3.8
如果您的 Jetson 上安装了 Jetpack,则可以使用
apt-get
安装 Python 的 tensorrt
模块。
在我安装了 Jetpack 的 Jetson Xavier NX 上
$ python3 -c "import tensorrt as trt; print(trt.__path__)"
['/usr/lib/python3.8/dist-packages/tensorrt']
$ dpkg -S /usr/lib/python3.8/dist-packages/tensorrt
python3-libnvinfer: /usr/lib/python3.8/dist-packages/tensorrt
所以你应该安装
python3-libnvinfer
才能在 Python 中获取 tensorrt
模块。
apt-get install python3-libnvinfer