https://www.cnblogs.com/zongfa/p/13947071.html
我安装的显卡的驱动版本是455,使用nvidia-smi看到该驱动支持的最高cuda版本为11.1,因此可以安装cuda10.1
Ubuntu20.04自带的gcc版本为9.3,而cuda10.1不支持gcc-9,因此要手动安装gcc-7,命令如下:
sudo apt-get install gcc-7 g++-7
安装完gcc-7,系统中就存在两个版本的gcc(g++),因此要设置默认的gcc和g++,命令如下:
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
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++
去官网下载cuda10.1的安装文件 cuda下载。使用run文件安装(使用deb文件会自动安装显卡驱动)

安装界面启动,直接选择continue,然后accept。

到这一步时注意,因为我们已经有安装了驱动,所以这里使用空格选中driver,取消驱动安装。然后移动最下面install,回车进行安装。
安装好后,可发现有了/usr/local/cuda-10.1这个文件夹,也就是安装路径所在。根据安装完成的提示,需要进行相关路径配置。在~/.bashrc中进行添加即可
安装cudnn
下载cudnn。登录下载与cuda10.1对应的cudnn 7.6.5,选择cudnn library for linux。下载好后进行解压,将解压的 cuda/include/cudnn.h 复制到 /usr/local/cuda/include,cuda/lib64/下所有文件复制到 /usr/local/cuda/lib64 下,并为这些文件添加读取和执行权限:
sudo chmod 755 /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
