安装PyTorch的详细过程记录

安装PyTorch的详细过程记录

PyTorch是一个开源机器学习框架，提供了便捷的各类机器学习、深度学习工具，支持动态计算图，可以在GPU上进行模型训练。下面详细讲解如何安装PyTorch。

准备工作

在安装PyTorch前，需要安装Python解释器。模型训练涉及到大量数学计算，建议安装numpy及scipy等科学计算库。另外，安装GPU版本还需要安装相关的CUDA和cuDNN库。

安装步骤

1. 打开PyTorch官网

进入https://pytorch.org/官网。

1. 选择合适的版本

根据自己的需求选择适合的版本，如CPU版本或GPU版本，推荐选择pip安装方式。

1. 复制命令

打开终端，复制对应版本的安装命令，如CPU版本的命令为：

pip install torch

GPU版本的命令为：

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

1. 安装依赖包

如果需要安装其他依赖包可以使用pip命令进行安装，如安装numpy和scipy：

pip install numpy
pip install scipy

1. 验证安装

打开python交互式环境，在命令行中输入import torch，如果没有报错，说明安装成功。

示例说明

例1：安装PyTorch CPU版本

1. 打开PyTorch官网，进入https://pytorch.org/

2. 选择pip安装方式，点击“Get Started”

3. 选择合适的版本，如CPU版本

4. 复制安装命令

pip install torch

1. 安装依赖包

pip install numpy
pip install scipy

1. 验证安装，打开python交互式环境

python
import torch

如果没有报错，说明安装成功。

例2：安装PyTorch GPU版本

1. 打开PyTorch官网，进入https://pytorch.org/

2. 选择pip安装方式，点击“Get Started”

3. 选择合适的版本，如CUDA 11.1版本的GPU版本

4. 复制安装命令

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

1. 安装依赖包

pip install numpy
pip install scipy

1. 验证安装，打开python交互式环境

python
import torch
print(torch.cuda.is_available())

如果输出True，说明已成功安装GPU版本。