怎么安装torch 1.7.0

以下是在 Linux 或 Windows 系统中安装 PyTorch 1.7.0 的详细步骤：

1. 安装 Anaconda 或 Miniconda

2. 在终端中执行以下命令创建一个名为“pytorch”环境，并安装 Python 3.6 版本：

conda create --name pytorch python=3.6

3. 激活新创建的“pytorch”环境：

conda activate pytorch

4. 安装 PyTorch 1.7.0（根据所需的 CUDA 版本选择对应的命令）

使用 CPU 版本：

conda install pytorch==1.7.0 torchvision==0.8.1 cpuonly -c pytorch

使用 CUDA 10.1 版本：

conda install pytorch==1.7.0 torchvision==0.8.1 cudatoolkit=10.1 -c pytorch

使用 CUDA 10.2 版本：

conda install pytorch==1.7.0 torchvision==0.8.1 cudatoolkit=10.2 -c pytorch

5. 安装其他需要的库

例如，安装 matplotlib：

conda install matplotlib

6. 测试 PyTorch 是否安装成功

在 Python 环境中输入以下代码：

import torch
print(torch.__version__)

如果输出的版本号为 1.7.0，则安装成功。