tensorrt安装
时间: 2023-08-08 19:11:35 浏览: 148
要安装TensorRT,可以使用pip安装或者使用deb包安装。
使用pip安装的步骤如下:
1. 首先,进入TensorRT的python目录:cd TensorRT-5.0.2.6/python
2. 然后,执行以下命令安装TensorRT:pip install tensorrt-5.0.2.6-py2.py3-none-any.whl
3. 接着,进入TensorRT的uff目录:cd TensorRT-5.0.2.6/uff
4. 执行以下命令安装UFF:pip install uff-0.5.5-py2.py3-none-any.whl
5. 最后,进入TensorRT的graphsurgeon目录:cd TensorRT-5.0.2.6/graphsurgeon
6. 执行以下命令安装graphsurgeon:pip install graphsurgeon-0.3.2-py2.py3-none-any.whl
如果想使用deb包安装,可以按照以下步骤进行:
1. 首先,下载对应系统版本的deb包,并记录下下载的版本号。
2. 执行以下命令安装deb包:sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_amd64.deb
3. 导入密钥:sudo apt-key add /var/nv-tensorrt-repo-${tag}/7fa2af80.pub
4. 更新软件源:sudo apt-get update
5. 安装TensorRT:sudo apt-get install tensorrt
6. 安装python-libnvinfer-dev(适用于Python2):sudo apt-get install python-libnvinfer-dev
7. 安装python3-libnvinfer-dev(适用于Python3):sudo apt-get install python3-libnvinfer-dev
8. 安装uff-converter-tf(适用于TensorFlow):sudo apt-get install uff-converter-tf
需要注意的是,导入uff模块时需要安装tensorflow模块,并且tensorflow的版本要与cuda版本对应。具体的版本对应关系可以参考官方网站。[1][2][3]
阅读全文