PyTorch Pip 安装教程

PyTorch Pip 安装教程

PyTorch 是一个开源的深度学习框架，由 Facebook 发布，可以用于构建各种深度学习模型。在使用 PyTorch 之前，你需要先安装它。本文将详细介绍如何使用 pip 在各种操作系统上安装 PyTorch。

安装前的准备

在安装 PyTorch 之前，你需要先安装 Python 和 pip。一般情况下，Python 3.6 及以上的版本已经自带了 pip。你可以通过以下命令检查 Python 和 pip 的安装情况：

python --version
pip --version

如果输出的信息中包含了 Python 和 pip 的版本号，那么说明你已经安装成功了。如果没有安装，你可以去 Python 官网下载安装包进行安装。

在 Linux 上安装 PyTorch

使用 pip 安装

在 Linux 上使用 pip 安装 PyTorch 非常简单，只需要执行以下命令：

pip install torch torchvision

这条命令会自动安装最新版本的 PyTorch 和 torchvision。安装完成后，你可以通过以下命令检查 PyTorch 是否成功安装：

python -c "import torch; print(torch.__version__)"

如果输出了 PyTorch 的版本号，那么说明安装成功了。

通过源码安装

如果你想要使用源码安装 PyTorch，可以通过以下步骤完成：

1. 克隆 PyTorch 仓库：

git clone --recursive https://github.com/pytorch/pytorch

2. 进入 PyTorch 目录，并安装依赖包：

cd pytorch
pip install -r requirements.txt

3. 安装 PyTorch：

python setup.py install

在 macOS 上安装 PyTorch

使用 pip 安装

在 macOS 上安装 PyTorch 也非常简单，只需要执行以下命令：

pip install torch torchvision

安装完成后，你可以通过以下命令检查 PyTorch 是否成功安装：

python -c "import torch; print(torch.__version__)"

使用 Homebrew 安装

你也可以使用 Homebrew 安装 PyTorch，执行以下命令：

brew install libomp
pip3 install torch torchvision

在 Windows 上安装 PyTorch

使用 pip 安装

在 Windows 上使用 pip 安装 PyTorch 也非常简单，只需要执行以下命令：

pip install torch torchvision

安装完成后，你可以通过以下命令检查 PyTorch 是否成功安装：

python -c "import torch; print(torch.__version__)"

使用 Anaconda 安装

如果你使用 Anaconda，在 Anaconda Prompt 中执行以下命令也可以安装 PyTorch：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

检查安装情况

无论是在 Linux、macOS 还是 Windows 上安装 PyTorch，你都可以通过以下命令来检查 PyTorch 是否成功安装：

import torch
print(torch.__version__)

如果运行结果显示了 PyTorch 的版本号，那么说明你已经成功安装 PyTorch。

总结

通过本文的介绍，你已经学会了如何使用 pip 在不同操作系统上安装 PyTorch。