安装前准备

确保您的系统已安装 Python（推荐 3.7-3.10 版本）和 pip 工具。

  • Windows 用户：建议使用 Anaconda 管理环境
  • MacOS/Linux 用户：可直接通过终端操作
  • 📌 注意：安装前请检查 GPU 驱动是否满足要求（如需 CUDA 支持）

安装方法

方法一：使用 pip（推荐）

pip install torch torchvision torchaudio

📌 若需指定版本： 

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html
PyTorch_Installation

方法二：使用 conda

conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch

📌 conda 安装会自动处理依赖关系

Conda_Installation

方法三：从源码编译

git clone https://github.com/pytorch/pytorch.git
cd pytorch
git checkout v1.13.1
python setup.py install

📌 需要较长时间，建议仅在特殊需求时使用

Source_Code_Installation

验证安装

运行以下代码检查是否成功：

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

✅ 若输出版本号且显示 True，说明安装成功！

扩展阅读

常见问题

⚠️ 安装失败？请检查：

  1. 网络是否通畅（可尝试更换镜像源）
  2. 系统架构是否匹配（如 x86_64 或 ARM64）
  3. 管理员权限是否开启（某些系统需要 sudo）
Installation_Troubleshooting