注:这里仅记录用tar安装tensorRT的流程,注意的是解压版的tar需要我们手动去修改lib的路径,如果是使用安装版的,就不需要了,安装版需要root权限进行安装。
1.查询自己的ubuntu、CUDA、cuDNN版本:
# 查看cuda版本
cat /usr/local/cuda/version.txt
# 查看cuDNN版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
# 查看ubuntu版本
cat /etc/issue
- 根据上面的三个版本,下载对应的TensorRT的tar包。
官网下载地址:https://developer.nvidia.com/nvidia-tensorrt-download
- 下载完后,安装:
# 在home下将下载的压缩文件进行解压
tar x