选择适合你的安装方式

根据你的开发环境和需求,选择以下安装方法:

使用 pip 安装

pip install torch torchvision torchaudio

📌 提示:确保已安装Python 3.8+,并根据你的CUDA版本选择对应版本的PyTorch。

PyTorch_logo

使用 conda 安装

conda install pytorch torchvision torchaudio -c pytorch

📌 提示:Anaconda用户可直接通过此命令安装,支持更便捷的环境管理。

Conda_environment

从源代码编译安装

适合需要自定义配置的高级用户:

  1. 安装PyTorch官方依赖
  2. 运行 setup.py 或使用 PyTorch构建工具
Source_code_compilation

验证安装

打开终端执行:

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

✅ 若输出版本号和True,表示安装成功!

常见问题

推荐阅读

安装完成后,可前往PyTorch官方文档获取更多API信息 📚