Windows 系统换源安装 PyTorch 超详细指南

一、为啥要换源安装 PyTorch

在 Windows 系统上安装 PyTorch 时，很多朋友会遇到下载速度慢的问题。这是因为默认的源服务器可能在国外，网络传输有延迟。换源就像是给下载找了一条更近、更畅通的路，能大大加快安装速度，让我们能更顺利地开启 PyTorch 的学习和使用之旅。

二、前期准备

1. 确认 Python 环境：PyTorch 依赖 Python，确保你已经安装了合适版本的 Python，建议安装 Python 3.7 及以上版本。你可以在命令提示符中输入 python --version 来查看已安装的 Python 版本。如果没安装或者版本不符合要求，就去 Python 官网下载安装包进行安装。
2. 安装 pip：pip 是 Python 的包管理工具，一般安装 Python 时会自动安装。要是不确定有没有安装，可以在命令提示符输入 pip --version。如果没有安装，在 Python 安装目录下找到 Scripts 文件夹，运行 get - pip.py 脚本就能安装。

三、选择合适的源

1. 常用国内源：国内有几个不错的源，比如清华大学的源、阿里云的源等。这些源不仅速度快，而且稳定性也不错。清华源的网址是 https://pypi.tuna.tsinghua.edu.cn/simple，阿里云源的网址是 https://mirrors.aliyun.com/pypi/simple。
2. 临时换源：临时换源很方便，在安装 PyTorch 的命令中加上 -i 参数和源的网址就行。比如用清华源安装 PyTorch，命令就是 pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple。这种方式只对当前这次安装有效。
3. 永久换源：想一劳永逸地使用某个源，可以在用户目录下创建一个 pip 文件夹（如果没有的话），在里面新建一个 pip.ini 文件（如果是 Windows 10，没有这个文件就新建一个文本文件，然后把后缀改成.ini）。在 pip.ini 文件里输入以下内容：

   

   [global]
   index - url = https://pypi.tuna.tsinghua.edu.cn/simple

   保存后，以后用 pip 安装包就会默认使用这个源了。

四、安装 PyTorch

1. 确定 CUDA 版本：如果你的电脑有 NVIDIA 显卡，并且想用 GPU 加速 PyTorch 运算，那得先确定电脑上安装的 CUDA 版本。可以在 NVIDIA 控制面板里查看，也能在命令提示符输入 nvcc -V 查看。要是没装 CUDA 或者想用 CPU 版本的 PyTorch，也没关系，安装命令稍有不同而已。
2. 安装命令：
   • CPU 版本：在命令提示符输入 pip install torch torchvision torchaudio，要是之前设置了永久换源，它会从设置的源下载安装包；要是没设置，也可以用临时换源的方式，比如 pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple。
   • GPU 版本：假设你的 CUDA 版本是 11.1，就在命令提示符输入 pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html。这里的版本号要根据实际的 CUDA 版本和你想用的 PyTorch 版本调整，具体对应关系可以去 PyTorch 官网查看。

五、验证安装是否成功

安装好后，打开 Python 交互式环境，在命令提示符输入 python 进入。然后依次输入以下代码：

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

如果没报错，并且能正常输出版本号，对于 GPU 版本还能正确显示 CUDA 是否可用，那就说明 PyTorch 安装成功啦。

按照上面的步骤，在 Windows 系统上换源安装 PyTorch 是不是也不难呀？赶紧试试，开启你的深度学习之旅吧！