在Ubuntu中安装TensorRT的步骤如下:
-
下载TensorRT安装包:首先需要从NVIDIA官网下载适用于Ubuntu的TensorRT安装包。下载完成后,解压安装包。
-
安装依赖库:在终端中输入以下命令安装TensorRT所需的依赖库:
sudo apt-get install libnvinfer7 libnvinfer-plugin7 libnvparsers7 libnvonnxparsers7 libnvinfer-bin7 libnvparsers7 libnvinfer-plugin7
- 安装TensorRT:在终端中切换到TensorRT安装包的目录,执行以下命令进行安装:
sudo dpkg -i nv-tensorrt-*.deb
- 配置环境变量:将TensorRT的库路径添加到LD_LIBRARY_PATH中,编辑~/.bashrc文件并添加以下内容:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/x86_64-linux-gnu/
- 重启终端:保存~/.bashrc文件并执行以下命令使配置生效:
source ~/.bashrc
安装完成后,您可以通过运行以下命令验证TensorRT是否已成功安装:
dpkg -l | grep TensorRT
如果输出中包含TensorRT的相关信息,则表示TensorRT已成功安装在Ubuntu系统中。