服务器 
首页 > 服务器 > 浏览文章

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

(编辑:jimmy 日期: 2024/9/24 浏览:3 次 )

CUDA安装下载cuda

输入nvidia-smi命令查看支持的cuda版本

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

如果无法查看,则说明尚未安装nvidia驱动,点击附加驱动,选择对应版本的驱动即可自动下载。

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

从https://developer.nvidia.com/cuda-toolkit-archive下载对应版本的cuda,选择18.04的版本即可。

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

gcc降级

Ubuntu20.04自带的gcc版本为9.7.0,需要添加gcc7才可安装cuda10.2,输入命令安装gcc7

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

查看gcc版本,可以看到目前系统中存在7和9两个版本

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

使用update-alternatives进行版本切换,输入以下命令:

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 50


此时输入sudo update-alternatives --config gcc命令查看gcc的默认版本,可以看到当前默认gcc版本为7,即切换成功。

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

安装cuda

输入命令进行安装

sudo sh cuda_10.2.89_440.33.01_linux.run

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

点击continue

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

输入accept

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

cuda安装包是自带显卡驱动的,所以这一步按空格去掉安装显卡驱动的选项,然后选择install

配置环境变量

输入gedit ~/.bashrc命令打开文件,在文件结尾输入以下语句,保存。

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

更新环境变量配置

source ~/.bashrc

至此cuda安装完成,输入nvcc -V命令查看cuda信息

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

安装cuDNN

从https://developer.nvidia.com/cudnn下载相应版本的cudnn,需要登陆。

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

下载解压之后,将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹,将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中,并添加读取权限:

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

至此cuDNN安装完成

上一篇:在Linux系统中使用Vim读写远程文件的命令详解
下一篇:详解ubuntu 20.04 LTS安装记录
一句话新闻
一文看懂荣耀MagicBook Pro 16
荣耀猎人回归!七大亮点看懂不只是轻薄本,更是游戏本的MagicBook Pro 16.
人们对于笔记本电脑有一个固有印象:要么轻薄但性能一般,要么性能强劲但笨重臃肿。然而,今年荣耀新推出的MagicBook Pro 16刷新了人们的认知——发布会上,荣耀宣布猎人游戏本正式回归,称其继承了荣耀 HUNTER 基因,并自信地为其打出“轻薄本,更是游戏本”的口号。
众所周知,寻求轻薄本的用户普遍更看重便携性、外观造型、静谧性和打字办公等用机体验,而寻求游戏本的用户则普遍更看重硬件配置、性能释放等硬核指标。把两个看似难以相干的产品融合到一起,我们不禁对它产生了强烈的好奇:作为代表荣耀猎人游戏本的跨界新物种,它究竟做了哪些平衡以兼顾不同人群的各类需求呢?