ubuntu安装tenserRT
时间: 2023-10-20 17:35:50 浏览: 43
要在Ubuntu上安装TensorRT,可以按照以下步骤进行操作:
1. 首先,确保你的Ubuntu系统已经安装了NVIDIA GPU驱动程序。你可以通过运行`nvidia-smi`命令来检查驱动程序是否正确安装。
2. 下载TensorRT的安装包。你可以从NVIDIA Developer网站上获取适合你系统的TensorRT版本。
3. 打开终端,进入到你下载的TensorRT安装包所在的目录,并解压缩安装包文件。
4. 进入解压缩后的文件夹,并执行安装脚本。你可以使用以下命令来运行脚本:
```
sudo ./<安装脚本文件名>.run
```
5. 在安装过程中,你需要接受许可协议以及选择安装的组件。根据你的需求进行选择。
6. 安装完成后,配置环境变量。在终端中打开`~/.bashrc`文件,添加以下行:
```
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-<版本号>/lib64
export PATH=$PATH:/usr/local/cuda-<版本号>/bin
```
注意替换`<版本号>`为你的CUDA版本号。
7. 保存`.bashrc`文件并执行以下命令来使环境变量生效:
```
source ~/.bashrc
```
8. 安装完成后,你可以在终端中运行`tensorrt`命令来验证TensorRT是否正确安装。
你现在应该已经成功在Ubuntu上安装了TensorRT。你可以根据你的具体需求使用TensorRT进行深度学习加速和推理。