Pytorch安装教程

PyTorch是一个流行的深度学习框架，它易于上手且功能强大。以下是安装PyTorch的基本步骤：

1. **检查Python版本**：首先确保你已经安装了Python 3.6或更高版本。

2. **选择安装包**：由于PyTorch有CPU和GPU两个版本，如果你的电脑支持CUDA（NVIDIA GPU），推荐安装 cudnn 加速版，否则选默认的CPU版本。

3. **安装anaconda**：Anaconda是Python的一个发行版，包含了许多科学计算库，包括PyTorch。你可以在官网下载适合你的操作系统（Windows, macOS, Linux）的安装包。

4. **创建新环境**：打开Anaconda Prompt（Windows）或终端（macOS/Linux），然后运行：

   conda create -n pytorch_env torch torchvision cpuonly

如果你想安装GPU版本，替换`cpuonly`为`pytorch torchvision cudatoolkit=10.2`（这里假设你的显卡兼容CUDA 10.2）。

5. **激活环境**：

   conda activate pytorch_env

6. **安装完成**：最后，确认安装是否成功，通过导入torch模块看看有没有错误。

7. **验证**：

   import torch
   print(torch.__version__)

如果你遇到任何问题，可以访问PyTorch官方文档（https://pytorch.org/get-started/locally/）获取更多帮助。