ubuntu tensorrt安装
时间: 2023-09-28 16:09:32 浏览: 62
要在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版本。
相关问题
ubuntu tensorrt安装 cuda11.6
要在Ubuntu上安装TensorRT和CUDA 11.6,你可以按照以下步骤进行操作:
1. 首先,你需要安装nv-tensorrt-repo的deb文件。使用以下命令将文件安装到系统中:
```
sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_amd64.deb
```
2. 安装TensorRT的密钥。使用以下命令将密钥添加到系统中:
```
sudo apt-key add /var/nv-tensorrt-repo-${os}-${tag}/*.pub
```
3. 更新apt源并安装TensorRT:
```
sudo apt-get update
sudo apt-get install tensorrt
```
4. 如果你只需要TensorRT而不使用TensorFlow,你可以安装numpy等所需的Python库:
```
python3 -m pip install numpy
```
5. 安装TensorRT的开发库:
```
sudo apt-get install python3-libnvinfer-dev
sudo apt-get install onnx-graphsurgeon
sudo apt-get install tensorrt-libs
sudo apt-get install tensorrt-dev
```
6. 如果你想保持旧版CUDA与TensorRT关联,可以使用以下命令:
```
sudo apt-mark hold tensorrt-dev
```
如果以后想取消关联,可以运行以下命令:
```
sudo apt-mark unhold tensorrt-dev
```
以上就是在Ubuntu上安装TensorRT和CUDA 11.6的步骤。希望对你有帮助!
ubuntu下tensorrt安装
TensorRT是英伟达推出的一个高性能深度学习推理库,可用于优化训练后的深度学习模型以在NVIDIA GPU上进行加速推理。在Ubuntu下安装TensorRT的步骤如下:
1. 下载TensorRT安装包并解压缩,可以通过官网下载,链接为:https://developer.nvidia.com/nvidia-tensorrt-download
2. 进入解压缩后的文件夹,运行以下命令:
```
$ cd TensorRT-<version>/python
$ sudo pip3 install tensorrt-<version>.whl
$ sudo pip3 install graphsurgeon-<version>.whl
```
其中,`<version>`为TensorRT的版本号。
3. 设置环境变量,运行以下命令:
```
$ echo "export LD_LIBRARY_PATH=/usr/local/cuda-<version>/lib64:${LD_LIBRARY_PATH}" >> ~/.bashrc
$ source ~/.bashrc
```
其中,`<version>`为CUDA的版本号。
至此,在Ubuntu下安装TensorRT完成。