Ubuntu20.04安装cuda10.1及对应版本的cuDNN7.6.5

一、安装前准备

首先要查看你的NVIDIA显卡驱动是否支持cuda10.1版本。
因为是在docker中执行，所以需要先进入你的容器：
nvidia-docker start 容器名
nvidia-docker attach 容器名
在你的容器里执行以下命令查看cuda版本：
nvidia-smi

CUDA Version: … 这里的版本号需要大于等于你安装的cuda版本号

二、关键点：gcc降级

因为Ubuntu20.04自带的gcc版本为9.3，而cuda10.1不支持gcc-9，因此要手动安装gcc-7，命令如下：

sudo apt-get install gcc-7 g+±7

安装完gcc-7，系统中就存在两个版本的gcc，因此要设置默认的gcc，命令如下：

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 1
此命令可以通过update-alternatives设置gcc各版本的优先级，优先级最高的为系统默认版本，可以用下述命令显示其优先级：
sudo update-alternatives --display gcc

设置默认的g++也是如此：

sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g+±7 9
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g+±9 1

显示g++优先级：

sudo update-alternatives --display g++

三、按照官网的安装步骤下载并安装CUDA 10.1

CUDA Toolkit 10.1 update2 Archive