TensorRT在Ubuntu上如何安装

751
2024/8/21 15:31:30
栏目: 智能运维
开发者测试专用服务器限时活动,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,并开始使用其强大的深度学习加速功能。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: Ubuntu下Scilab能顺利启动吗