Ubuntu18.04下安装CUDA10.2+cudnn8+TensorRT6.0

可分为四部分

• nvidia driver
• CUDA
• cudnn
• TensorRT

nvidia driver

两种方式，图形界面和命令行。

图形界面

找到下面的图标，打开

选择推荐的nvidia driver

（我已经安装了推荐的nvidia driver）

命令行

先看一下系统推荐的nvidia driver

ubuntu-drivers devices

输出类似这种

卸载之前的nvidia driver及其相关

sudo apt-get autoremove nvidia*

更新仓库列表

sudo apt-get update

安装推荐的nvidia driver及其相关

sudo apt-get install nvidia-settings nvidia-driver-440 nvidia-prime

CUDA

deb包下载

先看一下截至目前，nvidia driver与CUDA版本的对照表

去nvidia官网下载相应版本的CUDA
各版本CUDA

安装

由于我的nvidia driver版本是440，因此给出Ubuntu下CUDA10.2的安装
CUDA官网安装指导

配置环境变量

编辑~/.bashrc

gedit ~/.bashrc

在末尾加上

export CUDA_HOME=/usr/local/cuda 
export PATH=$PATH:$CUDA_HOME/bin 
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存退出，source一下

source ~/.bashrc

验证

nvcc -V

输出类似如下，即为配置成功

cudnn

deb包下载

各版本cudnn
根据我系统的情况，选择如下三个

安装

deb安装的不需要移动.h文件，安装

sudo dpkg -i libcudnn8_8.0.2.39-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.2.39-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn8-doc_8.0.2.39-1+cuda10.2_amd64.deb

网上有人说三个包的安装顺序不能错，没试过错了会怎样

TensorRT

TensorRT官网安装指导
通过阅读官网安装指导，我们发现（敲黑板）

所以说安装CUDA和cudnn时用deb包装会省掉可能出现一些问题

deb包下载

各版本TensorRT
根据我系统的情况，选择如下

安装

TensorRT官网安装指导-deb包安装部分
将自己对应的版本和系统号填进去

os="ubuntu1x04"
tag="cudax.x-trt7.x.x.x-ga-yyyymmdd"
sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_amd64.deb
sudo apt-key add /var/nv-tensorrt-repo-${tag}/7fa2af80.pub

sudo apt-get update
sudo apt-get install tensorrt cuda-nvrtc-x-y

验证

dpkg -l | grep TensorRT

输出类似如下，即为安装成功