pytorch安装教程

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

1. 安装 Python

在官网下载最新版 Python 安装包并进行安装。官网地址：https://www.python.org/downloads/windows/

2. 安装 PyTorch

可以通过以下两种方式安装 PyTorch：

方式一：使用 pip 安装

打开命令提示符窗口（Win+R，输入cmd），输入以下命令（根据您的 Python 版本和硬件环境选择相应的命令）：

- CPU 版本：pip install torch torchvision torchaudio
- GPU 版本：pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

方式二：使用 Anaconda 安装

首先，安装 Anaconda。官网地址：https://www.anaconda.com/products/individual

然后，打开 Anaconda Prompt（Win+R，输入anaconda prompt），输入以下命令：

- CPU 版本：conda install pytorch torchvision torchaudio cpuonly -c pytorch
- GPU 版本：conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch

3. 验证 PyTorch 安装

打开 Python 解释器或者 Jupyter Notebook，在命令行中输入以下代码：

import torch
print(torch.__version__)

如果输出正确的版本号，则说明 PyTorch 安装成功。

以上就是在 Windows 系统下安装 PyTorch 的步骤。如果您使用的是其他操作系统，可以参考官方文档进行安装。