乌班图系统安装tensorrt
时间: 2023-09-08 22:16:06 浏览: 121
对于在Ubuntu系统上安装TensorRT,可以按照以下步骤进行操作:
1. 首先,前往NVIDIA官方网站(https://developer.nvidia.com/tensorrt)下载适用于Ubuntu的TensorRT安装包。确保选择与您的系统版本和CUDA版本相匹配的正确版本。
2. 下载完成后,打开终端并进入下载目录。解压缩安装包,可以使用以下命令:
```
tar xzvf <下载的TensorRT安装包>.tar.gz
```
3. 进入解压缩后的目录,并执行安装脚本,例如:
```
cd <解压缩后的目录>
sudo ./install.sh
```
4. 安装过程中,您将被要求接受许可协议和选择安装路径。按照提示完成安装。
5. 安装完成后,您需要将TensorRT添加到系统环境变量中。打开终端,并编辑`~/.bashrc`文件(或者`~/.zshrc`文件),在文件末尾添加以下行:
```
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/TensorRT-<安装的版本号>/lib
export PATH=$PATH:/usr/local/TensorRT-<安装的版本号>/bin
```
然后保存文件并执行以下命令,使环境变量生效:
```
source ~/.bashrc
```
6. 现在,您已经成功安装了TensorRT。您可以使用TensorRT提供的Python API来开发和优化深度学习模型。
阅读全文