【TensorRT系列笔记-01】TensorRT 5.0.2.6 安装

官方下载地址:https://developer.nvidia.com/nvidia-tensorrt-5x-download

选择对应版本,根据你的OS以及CUDA选择对应的版本,如下图所示:

【TensorRT系列笔记-01】TensorRT 5.0.2.6 安装_第1张图片

 对下载好的文件进行更名,并解压;

tar -xzvf TensorRT-5.0.2.6.tar.gz

 【TensorRT系列笔记-01】TensorRT 5.0.2.6 安装_第2张图片

 配置环境变量:

vim ~/.bashrc

添加环境变量:

export LD_LIBRARY_PATH=TensorRT路径/lib:$:$LD_LIBRARY_PATH

进行生效:

source ~/.bashrc

 测试:在samples中提供了很多官方的案例,我们首先将这些案例跑起来:

cd samples
make
cd ../bin
./sample_mnist

 运行结果如下:

【TensorRT系列笔记-01】TensorRT 5.0.2.6 安装_第3张图片

完,

你可能感兴趣的:(TensorRT)