官方下载地址:https://developer.nvidia.com/nvidia-tensorrt-5x-download
选择对应版本,根据你的OS以及CUDA选择对应的版本,如下图所示:
对下载好的文件进行更名,并解压;
tar -xzvf TensorRT-5.0.2.6.tar.gz
配置环境变量:
vim ~/.bashrc
添加环境变量:
export LD_LIBRARY_PATH=TensorRT路径/lib:$:$LD_LIBRARY_PATH
进行生效:
source ~/.bashrc
测试:在samples中提供了很多官方的案例,我们首先将这些案例跑起来:
cd samples
make
cd ../bin
./sample_mnist
运行结果如下:
完,