随着深度学习技术的不断发展，PyTorch 作为一种流行的深度学习框架，受到了众多开发者和研究人员的青睐。本文将为您详细讲解如何在 Windows 10 系统下安装 PyTorch。
一、准备工作
1. 确保您的 Windows 10 系统已更新至最新版本。
2. 准备 Python 环境和 pip 工具，确保 Python 版本为 3.5 以上。
3. 根据您的需求选择合适的 PyTorch 版本，包括 CPU 版本和 GPU 版本。
二、安装步骤
1. 访问 PyTorch 官方网站（https://pytorch.org/get-started/locally/），选择合适的安装方式。
2. 下载安装包：
- 如果您使用的是 CPU 版本，下载相应的 Python 包。
- 如果您使用的是 GPU 版本，下载包含 CUDA 支持的 Python 包。
3. 打开命令提示符（cmd）或 PowerShell，执行以下命令安装 PyTorch：
```bash
pip install torch==1.x.x torchvision==0.x.x torchaudio==0.x.x --extra-index-url https://download.pytorch.org/whl/torch_stable.html
```
其中，`1.x.x`、`0.x.x` 分别代表 PyTorch、torchvision 和 torchaudio 的版本号，您可以根据需要修改。
4. 安装完成后，验证安装是否成功：
```bash
python -c "import torch; print(torch.__version__)"
```
如果输出正确的版本号，说明 PyTorch 安装成功。
5. （可选）安装 CUDA 驱动程序：
- 访问 NVIDIA 官方网站（https://www.nvidia.com/Download/index.aspx）下载与您的 GPU 兼容的驱动程序。
- 安装驱动程序并重启计算机。
6. （可选）安装 CUDA Toolkit：
- 访问 NVIDIA CUDA Toolkit 官方网站（https://developer.nvidia.com/cuda-toolkit）下载与您的 CUDA 驱动程序兼容的 CUDA Toolkit。
- 解压下载的文件，进入安装目录。
- 执行以下命令安装 CUDA Toolkit：
```bash
vs2017\vcvarsall.bat x64
```
安装过程中，您需要选择合适的安装路径。
7. （可选）配置环境变量：
- 在系统属性中，选择“环境变量”选项卡。
- 在“系统变量”中，找到“Path”变量，点击“编辑”。
- 在变量值中添加 CUDA Toolkit 的安装路径，例如：`C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vX.Y.X\bin;`
- 在“用户变量”中，找到“Path”变量，点击“编辑”。
- 添加 CUDA Toolkit 的安装路径，例如：`C:\Users\YourUsername\AppData\Local\NVIDIA Corporation\CUDA\vX.Y.X\bin;`
- 保存并关闭所有窗口。
至此，您已成功在 Windows 10 系统下安装 PyTorch。接下来，您可以开始使用 PyTorch 进行深度学习开发了。
