【模型部署】CUDA 10.2 + cuDNN 8.1.1 + TensorRT 7.0.0 安装踩坑记录

0.本地环境说明：

• Ubuntu1804
• NVIDIA驱动440
• CUDA10.2
• cuDNN8.1.1
• TensorRT7.0.0

1.NVIDIA驱动版本查询

nvidia-smi

2.CUDA10.2

由于之前的CUDA安装的是10.1，所以需要更新（卸载重装）

2.1 卸载CUDA：

1. 如果之前是.run文件安装的话：

sudo /usr/local/cuda/bin/cuda-uninstaller

2. 如果之前是deb文件安装的话：

sudo dpkg -r xxxcudaxxx

2.2 安装CUDA10.2

注意！！！！这里有个坑

安装官方文档的第3点：

安装tensorRT如果采用deb包安装，那么CUDA也必须采用deb包安装

CUDA10.2下载
CUDA安装官方文档

CUDA安装可能会报依赖项错误，

用aptitude再试一遍就好

3.安装cuDNN8.1.1

cuDNN8.1.1下载
nvidia官方安装文档

4. 安装TensorRT

TensorRT下载
TensorRT官方安装文档

sudo dpkg -i 'nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.0.0.11-ga-20191216_1-1_amd64.deb' 
sudo apt-key add /var/nv-tensorrt-repo-cuda10.2-trt7.0.0.11-ga-20191216/7fa2af80.pub
sudo apt-get update
sudo apt-get install tensorrt

最后一行命令报错
sudo apt-get install tensorrt

最后是aptitude命令安装，yyds

sudo aptitude install tensorrt

拒绝第一个方案，选择第二个兼容方案

5.TensorRT验证

sudo dpkg -l | grep TensorRT

sudo dpkg -l | grep TensorRT

6.总结

1. sudo aptitude install xxx 牛逼
2. NVIDIA官方文档yyds！虽然不好找到文档的入口，而且全英文，但是只要仔细安装官方步骤来，可以走很少弯路。