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.cn
3 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版本即表示成功：
