tensorrt安装 ubuntu
时间: 2023-07-31 11:08:22 浏览: 58
根据引用[1]和引用[2]的内容,以下是在Ubuntu上安装TensorRT的步骤:
1. 根据你的CUDA版本,在官网上下载对应版本的TensorRT的tar包。
2. 解压tar包,并将解压后的TensorRT库路径添加到LD_LIBRARY_PATH环境变量中。
3. 进入TensorRT的python目录,使用pip安装对应版本的tensorrt包。
4. 验证TensorRT安装是否成功,可以运行Python代码`import tensorrt;print(tensorrt.__version__)`来输出TensorRT的版本号。
5. 安装uff和graphsurgeon,分别进入对应的目录,使用pip安装对应版本的whl文件。
根据引用[3]的内容,还需要进行以下准备工作:
1. 确保已经安装了Python,并且在TensorRT的python文件夹下有对应版本的whl文件。
2. 使用pip安装对应版本的whl文件,以添加对Python的支持。
3. 运行测试代码`import tensorrt as trt;print(trt.__version__)`来验证安装是否成功。
综上所述,你可以按照以上步骤在Ubuntu上安装TensorRT。
相关问题
ubuntu tensorrt安装
要在Ubuntu上安装TensorRT,您可以按照以下步骤进行操作:
1. 下载TensorRT的tar包,并解压缩。您可以使用命令`tar xzvf TensorRT-7.x.x.x.<os>.<arch>-gnu.cuda-x.x.cudnn7.x.tar.gz`来解压缩。请确保将`x.x.x.x`替换为您要安装的TensorRT版本,`<os>`替换为您的操作系统(Ubuntu-14.04,Ubuntu-16.04,Ubuntu-18.04,CentOS-7.6),`<arch>`替换为您的架构(x86_64或ppc64le),`cuda-x.x`替换为您的CUDA版本(9.0,10.0或10.1),`cudnn7.x`替换为您的cuDNN版本(7.6)。
2. 进入解压后的TensorRT目录,并安装TensorRT的Python接口。您可以使用命令`cd TensorRT-7.x.x.x/python`进入Python接口目录,然后使用命令`pip install tensorrt-7.x.x.x-cp3x-none-linux_x86_64.whl`安装TensorRT的Python接口。请确保将`7.x.x.x`替换为您要安装的TensorRT版本。
安装完成后,您就可以在Ubuntu上使用TensorRT了。请注意,您还需要确保您的系统满足TensorRT的要求,包括正确的操作系统版本,适当的GPU型号和驱动版本,以及所需的CUDA和cuDNN版本。
安装tensorrt ubuntu
可以通过以下步骤在 Ubuntu 上安装 TensorRT:
1. 下载 TensorRT 安装包:https://developer.nvidia.com/nvidia-tensorrt-download
2. 解压安装包并进入解压后的目录。
3. 运行以下命令安装 TensorRT:
```
sudo dpkg -i nv-tensorrt-*.deb
sudo apt-get update
sudo apt-get install tensorrt
```
4. 安装完成后,可以通过运行以下命令验证 TensorRT 是否已正确安装:
```
dpkg -l | grep TensorRT
```
如果输出类似于以下内容,则表示 TensorRT 已成功安装:
```
ii libnvinfer-dev 7.2.2-1+cuda11.1 amd64 TensorRT development libraries and headers
ii libnvinfer-plugin-dev 7.2.2-1+cuda11.1 amd64 TensorRT plugin libraries and headers
ii libnvinfer-plugin7 7.2.2-1+cuda11.1 amd64 TensorRT plugin libraries
ii libnvinfer7 7.2.2-1+cuda11.1 amd64 TensorRT runtime libraries
ii python3-libnvinfer 7.2.2-1+cuda11.1 amd64 Python 3 bindings for TensorRT
ii python3-libnvinfer-dev 7.2.2-1+cuda11.1 amd64 Python 3 development package for TensorRT
ii python3-libnvinfer-plugin 7.2.2-1+cuda11.1 amd64 Python 3 TensorRT plugin libraries
ii python3-libnvinfer-plugin-dev 7.2.2-1+cuda11.1 amd64 Python 3 development package for TensorRT plugin
ii python3-libnvinfer-dev 7.2.2-1+cuda11.1 amd64 Python 3 development package for TensorRT
ii tensorrt 7.2.2.3-1+cuda11.1 amd64 Meta package of TensorRT
```