欢迎来到 PyTorch 安装指南页面!以下内容将指导您如何安装 PyTorch,并开始使用这个强大的深度学习框架。
系统要求
在安装 PyTorch 之前,请确保您的系统满足以下要求:
- 操作系统:Linux、macOS 或 Windows
- Python 版本:3.6 或更高版本
安装步骤
以下是安装 PyTorch 的步骤:
安装 Python:如果您还没有安装 Python,请从 Python 官网 下载并安装最新版本的 Python。
安装 PyTorch:打开命令行工具,输入以下命令:
pip install torch torchvision torchaudio
如果您需要使用 GPU 加速,请使用以下命令:
pip install torch torchvision torchaudio torchvision-cuda
验证安装:安装完成后,可以通过以下命令验证 PyTorch 是否已正确安装:
import torch print(torch.__version__)
如果输出 PyTorch 版本号,则表示安装成功。
注意事项
- 在安装过程中,如果遇到任何问题,可以查阅 PyTorch 安装文档 获取帮助。
- 为了更好地使用 PyTorch,建议您阅读 PyTorch 教程 学习更多基础知识。
示例图片
下面是一张 PyTorch 图标的示例图片: