【亲测】ubuntu20.4安装TensorRT_ubuntu20.04 tensorrt8.6

作者：数据流舞者 | 2024-02-03 18:31:30

踩

ubuntu20.04 tensorrt8.6

安装TensorRT之前请确保显卡驱动 CUDA cuDNN全部成功安装

安装CUDA+显卡驱动请参照：

【亲测】ubuntu20.4 显卡驱动+CUDA11.8一起安装_Coding_C++的博客-CSDN博客

安装cuDNN请参照：

【亲测】ubuntu20.4 安装cuDNN v8.9.3_Coding_C++的博客-CSDN博客

2、选择TensorRT8

3、选择TensorTR8.6 GA

4、我们选择TAR格式安装（.deb安装很麻烦）

5、选择后会进行下载，直到下载完成。

1、下载完成后会在Downloads下有（如：TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-11.8.tar.gz）

2、解压 tar -xzvf TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-11.8.tar.gz

3、执行命令sudo cp -r TensorRT-8.6.1.6 / 将解压的内容拷贝到指定目录（我是到根目录了）

1、添加系统路径

sudo gedit ~/.bashrc

2、打开~/.bashrc最后添加

export LD_LIBRARY_PATH=/TensorRT-8.6.1.6/lib:$LD_LIBRARY_PATH

3、让添加的路径生效

source ~/.bashrc

1、在/TensorRT-8.6.1.6/samples/下找到sampleOnnxMNIST

2、进入sampleOnnxMNIST文件夹

cd sampleOnnxMNIST

3、编译sampleOnnxMNIST

4、编译成功后在/TensorRT-8.6.1.6/bin会生成sample_onnx_mnist

5、执行sample_onnx_mnist

6、成功后显示

至此TensorRT安装成功

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/article/detail/57585