TensorRT在Ubuntu上如何安装

avatar
作者
猴君
阅读量:0

要在Ubuntu上安装TensorRT,您可以按照以下步骤操作:

  1. 下载TensorRT的安装包,可以在NVIDIA官方网站上找到最新的版本。
  2. 安装依赖项,可以通过以下命令安装:
sudo apt-get update sudo apt-get install libnvinfer7 libnvonnxparsers7 libnvparsers7 libnvinfer-plugin7 libnvinfer-bin libnvparsers-dev libnvinfer-plugin-dev libnvonnxparsers-dev libnvinfer-dev 
  1. 解压缩TensorRT安装包:
tar xzvf <TensorRT安装包.tar.gz> 
  1. 进入TensorRT安装目录:
cd <TensorRT安装目录> 
  1. 运行安装脚本进行安装:
sudo ./TensorRT-<版本号>.<补丁号>.<补丁号>.<补丁号>-Linux-x86_64.run 
  1. 安装完成后,设置TensorRT的环境变量:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/x86_64-linux-gnu export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/aarch64-linux-gnu 
  1. 最后,您可以验证TensorRT是否成功安装:
dpkg -l | grep TensorRT 

通过以上步骤,您可以在Ubuntu上成功安装TensorRT,并开始使用其强大的深度学习加速功能。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!