PyTorch 安装指南

以下是安装 PyTorch 的步骤，确保您按照以下指南进行操作。

系统要求

• 操作系统：Linux、macOS 或 Windows
• Python 版本：Python 3.6-3.8

安装步骤

1. 安装 Python：确保您的系统已安装 Python 3.6-3.8 版本。
2. 安装 PyTorch：打开终端或命令提示符，执行以下命令：

pip install torch torchvision torchaudio

3. 验证安装：在终端或命令提示符中，运行以下命令来验证 PyTorch 是否已正确安装：

import torch
print(torch.__version__)

注意事项

• 如果您需要 GPU 支持，请使用以下命令安装 PyTorch：

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

• 请确保您的 CUDA 版本与 cudatoolkit 版本兼容。

扩展阅读

如果您想了解更多关于 PyTorch 的信息，请访问我们的PyTorch 教程页面。

图片示例

PyTorch Logo