2022-11-25
一、概述
计算机的一个重要组成部分是处理器,包括CPU和GPU,CPU包括算数运算和逻辑运算。
通常一个电脑都包含CPU,而不一定配备GPU。GPU更擅长图像处理及矩阵运算,包括目前比较火的机器学习都会用到GPU,当然CPU也可以,只是效率没有GPU高。
GPU的大公司主要是:英伟达、AMD、因特尔、ARM、高通Adreno、景嘉微电子、芯原微电子等等;
而目前主流显卡的显示芯片主要分为:英伟达NVIDIA 和 AMD;
我们这里以 英伟达NVIDIA 为例子。
二、NVIDIA显卡,显卡驱动和CUDA版本之间的关系
NVIDIA显卡:是一款硬件板子,主流是NVIDIA的GPU,深度学习本身需要大量计算。GPU的并行计算能力,在过去几年里恰当地满足了深度学习的需求。 显卡驱动:告诉计算机识别显卡硬件,调用其计算资源。 CUDA:是NVIDIA推出的只能用于自家GPU的并行计算框架。只有安装这个框架才能够进行复杂的并行计算。主流的深度学习框架也都是基于CUDA进行GPU并行加速的。还有一个叫做cudnn,是针对深度卷积神经网络的加速库。
具体操作步骤:
1、使用快捷键【Ctrl+Alt+T】打开终端;
2、输入以下命令查看显卡版本号。
# 查看显卡型号 lspci | grep -i nvidia # 这里以 nvidia 为例子 #或者 ubuntu-drivers devices
相关操作命令:
# 查看驱动版本 sudo dpkg --list | grep nvidia-* #查看发布版本号 cat /etc/issue lsb_release -a #查看内核版本号 uname -sr uname -a #查看CPU信息 cat /proc/cpuinfo | grep "physical id" | uniq | wc -l #查看CPU个数 cat /proc/cpuinfo | grep "cpu cores" | uniq #查看CPU核数 cat /proc/cpuinfo | grep 'model name' |uniq #查看CPU型号 #查看gpu信息 nvidia-smi #GPU列表信息 nvidia-settings #显示你的显卡信息 lshw -c video #查看显卡型号 lspci | grep -i nvidia #得到NVIDIA显卡的信息
接下来下载NVIDIA显卡驱动以及CUDA,他们之间有版本约束:
可以通过以下网址查看: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
NVIDIA显卡驱动下载:
输入自己的信息,包括产品类型、系列、产品、操作系统、语言等,然后点击开始搜索,结果在下面:
搜索结果如下:
根据实际情况选择具体的版本。
CUDA下载:
下载官网:
https://developer.nvidia.com/cuda-toolkit-archive
https://developer.nvidia.com/cuda-downloads
根据情况选择版本及操作系统等。