win10 conda 环境下安装 pytorch 和相关工具集

图形处理器($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$ 的下载范围。

    

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

    

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

  

很多基于 $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()

可以看到