windows 安装pytorch

安装 PyTorch 前，首先需要安装 Python。建议安装 Python 3.7 或更高版本。可以从官方网站 https://www.python.org/downloads/ 下载安装程序。

安装 PyTorch 有两个版本：CPU 版本和 GPU 版本。CPU 版本只能利用 CPU 进行计算，而 GPU 版本支持使用 NVIDIA 显卡进行计算，速度更快。如果你的机器有 NVIDIA 显卡并且安装了 CUDA，建议安装 GPU 版本。

以下是在 Windows 上安装 PyTorch 的步骤：

1. 打开 Windows Powershell 或者 CMD。

2. 安装 Anaconda，可以从官网 https://www.anaconda.com/products/individual 下载安装程序。

3. 创建一个新的 Conda 环境：

conda create --name pytorch_env

4. 激活环境：

conda activate pytorch_env

5. 安装 PyTorch，可以通过 pip 安装：

pip install torch torchvision torchaudio

如果需要安装 GPU 版本，可以在上面的命令后面加上 `-f https://download.pytorch.org/whl/cu111/torch_stable.html`。

6. 安装完毕后，可以通过以下代码测试是否安装成功：

import torch

if torch.cuda.is_available():
    device = torch.device("cuda")
else:
    device = torch.device("cpu")

print(device)

运行以上代码后，如果输出结果为 `cuda`，说明 PyTorch 安装成功并且能够使用 GPU 进行计算。如果输出结果为 `cpu`，说明安装的是 CPU 版本的 PyTorch。

希望这能帮到你。