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

最新推荐文章于 2024-09-20 20:38:13 发布

原创

最新推荐文章于 2024-09-20 20:38:13 发布 · 2.7k 阅读

15 ·

CC 4.0 BY-SA版权

文章标签：

#其他

本文详细介绍了如何在Ubuntu20.04上安装CUDA 10.1及对应的cuDNN 7.6.5。内容包括检查NVIDIA驱动、降级gcc、按照官方步骤安装CUDA、解决安装过程中遇到的问题、配置环境变量以及多版本CUDA的切换和cuDNN的下载与安装。

一、安装前准备

首先要查看你的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++