1 概述
Windows下 Python+CUDA+PyTorch 装置，步骤都很具体，特此记录下来，帮忙读者少走弯路。
2 Python
Python的装置还是比较简单的，从官网下载 exe 安装包即可：
因为目前最新的 torch 版本只反对到Python 3.8，因而为了不会呈现版本兼容问题倡议装置Python 3.8。
下载两头的那个 executable installer 并装置，关上会倡议抉择自定义：
抉择须要的组件：
抉择地位：
装置完后能够从命令行查看是否装置胜利（个别默认会把门路增加到环境变量中）：
python --version
因为该安装包会主动把 pip 也一起装置，因而在应用时会提醒 pip 须要降级，请应用如下命令降级pip：
python -m pip install --upgrade pip
在装置前倡议增加国内源，在用户目录下新建一个 pip 文件夹，并在其中新建一个叫 pip.ini 的文件，输出如下内容：
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host = https://pypi.tuna.tsinghua.edu.cn3 CUDA
CUDA的装置也不难，个别 Windows 会在装零碎的时候主动装置好对应的 N 卡 驱动，从官网下载对应的CUDA Toolkit：
下载好后关上（实际上是一个压缩包，这里是解压门路）：
解压实现后会主动关上装置界面：
没非凡需要能够间接应用默认装置选项，否则请抉择自定义组件进行装置：
因为笔者没有装置VS，这里会提醒找不到VS，疏忽即可：
接下来就是期待装置：
装置实现：
关上 cmd 输出nvcc --version，显示版本号即为装置胜利：
4 PyTorch
4.1 下载
PyTorch倡议应用离线形式手动装置，戳这里下载对应 CUDA 版本与对应 Python 版本的 torch 以及torchvision：
最后面的 cu110 示意 cuda 11.0，接着是torch/torchvision，再接着是版本，前面的cp 示意 Python 版本，比方 cp36 示意Python 3.6，接着是零碎和架构。
4.2 装置
下载实现后应用 pip 装置即可：
pip install torch-1.7.0+cu110-cp38-cp38-win_amd64.whl
pip install torchvision-0.8.1+cu110-cp38-cp38-win_amd64.whl
留神如果是 Windows 2004 版本可能会呈现如下问题：
这是 numpy 19.0.4 版本的问题，详情能够戳这里。
解决办法是装置 19.0.3 版本：
pip uninstall numpy
pip install numpy==19.0.3接着进行测试，关上python：
python
import torch
print(torch.__version__)输入 torch 版本即示意胜利：