可分为四部分
- 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官网安装指导
通过阅读官网安装指导,我们发现(敲黑板)
所以说用deb或rpm包安装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
输出类似如下,即为安装成功