在腾讯云服务器安装tensorrt8.0（cuda11.0.3，cudnn8.1，ubuntu -server18.04）

最新推荐文章于 2023-10-19 16:47:37 发布

Funny_AI_LAB

最新推荐文章于 2023-10-19 16:47:37 发布

阅读量1k

点赞数 1

分类专栏： tensorRT 文章标签： ubuntu 服务器腾讯云

本文链接：https://blog.csdn.net/sexy19910923/article/details/123135036

版权

tensorRT 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文详细介绍了如何在Ubuntu系统上查看CUDA版本、CUDNN版本，并通过官方途径下载和安装TensorRT，包括.deb包和.tar包的方式，以及Python集成验证。涵盖从环境配置到验证步骤的全过程，适合TensorRT新手和开发者参考。

摘要由CSDN通过智能技术生成

查询服务器的相关信息：
查看cuda版本：cat /usr/local/cuda/version.txt
在这里插入图片描述
查看cudnn的版本

去nvidia官网找对应的tensorrt版本：https://developer.nvidia.com/nvidia-tensorrt-8x-download
两种方式安装：
一.deb的版本
1.下载deb包去官网

os="ubuntu1x04"
tag="cudax.x-trt7.x.x.x-ga-yyyymmdd"
sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_amd64.deb
sudo apt-key add /var/nv-tensorrt-repo-${tag}/7fa2af80.pub

sudo apt-get update
sudo apt-get install tensorrt cuda-nvrtc-x-y

如果使用 Python 3.x：
sudo apt-get install python3-libnvinfer-dev

2.将安装以下附加软件包：
python3-libnvinfer

3.如果您打算将TensorRT与TensorFlow一起使用：
sudo apt-get install uff-converter-tf
这 graphsurgeon-tf包也将使用上述命令安装。
4.验证安装
dpkg -l | grep TensorRT``
或者进入python环境
import tensor
print（tensorrt.__version__）

二.下载tar包
根据你的cuda，cudnn环境下载对应的tar包
我下载的TensorRT-8.0.1.6.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz

1.解压tar
tar -xzvf TensorRT-8.0.1.6.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz
2.添加绝对路径
vim ~/.bashrc
export LD_LIBRARY_PATH=/home/ubuntu/tensorrt8/TensorRT-8.0.1.6/lib:$LD_LIBRARY_PATH
source ~/.bashrc

3.安装tensorrt
cd python
pip install tensorrt-8.0.1.6-cp38-none-linux_x86_64.whl

4.安装uff
cd uff/
pip install uff-0.6.9-py2.py3-none-any.whl
# 验证uff
which convert-to-uff

5.安装graphsurgeon
cd graphsurgeon
pip install graphsurgeon-0.4.5-py2.py3-none-any.whl