Ubuntu16.04 安装TensorRT遇到的坑小结
1.如果想安装tensorRT的deb版本,cuda必须使用deb安装
然而cuda的deb版本安装的时候可能会覆盖你已经安装的nvidia驱动,建议先卸载了nvidia驱动,再执行cuda.deb包的安装,完成安装后会自动给你安装合适的nvidia驱动。
2.安装过程遇到E: Unable to locate package tensorrt
解决方法:网上的换源、sudo apt-get update可能无法解决问题
我试过TensorRT7.0.0.1、TensorRT7.2.3.4,换了无数国内源仍然还是Unable to locate package tensorrt,最后的解决方案是换到TensorRT7.1.3.4版本才解决问题(我估计国内源最新或比较老老版本都没有保留的原因吧):
nv-tensorrt-repo-ubuntu1604-cuda10.2-trt7.1.3.4-ga-20200617_1-1_amd64.deb
想要确定sudo apt-get update是否成功,可以这样做:
$ cd /etc/apt/sources.list.d
$ ls
如果看到nv-tensorrt-cuda10.2-trt7.1.3.4-ga-20200617.list就证明可以安装!