图形处理器($Graphics \; Processing \; Unit$,即 $GPU$),又称显卡、视觉处理器、显示芯片,是一种专门在个人电脑、工作站、游戏机和一些移动设备上图

像运算工作的微处理器。它是一个硬件,是外部的设备,自然需要一套软件来让 $CPU$ 操作 $GPU$。这个软件程序就叫显卡驱动。软件需要编译,自然高

度依赖操作系统,因此即使同样型号的显卡在不同操作系统一般需要的驱动也不一样。

程序员也可以直接写代码与显卡驱动对接,从而直接控制 $GPU$ 处理器,但是驱动程序操作起来实在太麻烦,于是 $NVIDIA$ 推出一套软件接口来简化

$GPU$ 操作,这便是 $CUDA$,它和显卡驱动是两回事。

$cudatoolkit$ 提供了 $CUDA$ 的开发环境,安装这个自然也就安装了 $CUDA$,工具包中还包含多个 $GPU$ 加速库、多种调试和优化工具、一个 $C/C++$

编译器 $nvcc$ 以及一个用于在主要架构,包括 $x86$、$Arm$ 和 $POWER$ 上构建和部署应用的运行时库。说白了就是一些编译器和库的集合,集成了一些

功能,方便用户在 $GPU$ 上做开发。

查看本机 $Nvidia$ 支持的 $CUDA$ 版本,可以看到这个位置对应的版本为 $11.0.208$,因为 $GPU$ 支持的 $CUDA$ 版本是向下兼容的(可能有些功能没法用),  

所以我们可以知道该电脑显卡支持的 $CUDA$ 版本不能超过 $11.0.208$,这样就确定了 $CUDA$ 的下载范围。

    

gpu 驱动模块架构 gpu驱动是什么意思_CUDA

再查看一下本机的显卡驱动,这里为 $451.67$ 版本

    

gpu 驱动模块架构 gpu驱动是什么意思_官网_02

然后通过网址可以得到显卡驱动和与之兼容的 $CUDA$ 版本,发现我的电脑显卡驱动版本还挺高。

  

gpu 驱动模块架构 gpu驱动是什么意思_官网_03

很多基于 $Pytorch$ 的工具集都非常好用,比如处理自然语言的 $torchtext$,处理音频的 $torchaudio$,以及处理图像视频的 $torchvision$。

下面我们进行安装:$pytorch$ $torchtext$ $torchaudio$ $torchvision$ $cuda10.2$。

配置文件 $.condarc$ 的内容如下所示,这些地址包含了需要的包,如果不清楚怎么配置,可先去阅读博客

ssl_verify: true
channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
show_channel_urls: true

官网提供了一个命令:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

安装命令中最后的 -c pytorch,是默认指定 anaconda 官网上的安装源,可能由于墙的问题,无法正常或者迅速下载。因为我这里选用清

华的源,所以需要去掉 -c pytorch。切换到需要安装的 conda 环境,执行:

conda install pytorch torchvision cudatoolkit=10.2

再装另外两个工具集(也可以放到上面那句命令里):

conda install torchtext torchaudio

安装完成后,输入 python 进入环境,然后依次输入

import torch
print(torch.__version__)
torch.cuda.is_available()

可以看到

    

gpu 驱动模块架构 gpu驱动是什么意思_显卡驱动_04