ubuntu20.04安装tensorRT
时间: 2023-11-06 11:05:30 浏览: 122
TensorRT安装
根据提供的引用内容,介绍在Ubuntu20.04上安装TensorRT的步骤如下:
1. 根据安装环境下cuda的版本号,在官网中下载对应版本的tar包,例如:TensorRT-7.2.2.3.Ubuntu-18.04.x86_64-gnu.cuda-11.0.cudnn8.0.tar.gz。
2. 解压tar包,可以使用命令:sudo tar -xzvf TensorRT-8.5.1.7.Linux.x86_64-gnu.cuda-11.8.cudnn8.6.tar.gz。
3. 添加环境变量,可以使用命令:export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/rjg/下载/TensorRT-8.5.1.7/lib ('注意/home/rjg/下载'是解压tar包的地址)。
4. 安装TensorRT,可以使用命令:cd TensorRT-8.5.1.7/python,然后使用pip安装tensorrt-8.5.1.7-cp38-none-linux_x86_64.whl。
5. 验证TensorRT安装是否成功,可以使用命令:python -c "import tensorrt;print(tensorrt.__version__)",若正确应输出版本号: 8.5.1.7。
6. 安装uff,可以使用命令:cd ../uff/,然后使用pip安装uff-0.6.9-py2.py3-none-any.whl。
7. 安装graphsurgeon,可以使用命令:cd ../graphsurgeon/,然后使用pip安装graphsurgeon-0.4.6-py2.py3-none-any.whl。
阅读全文