安装方法一览

以下是安装 PyTorch 教程的几种常用方式:

📦 方法一:使用 pip 安装

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118

✅ 适用于大多数 Python 环境
🔗 了解更多安装选项

🐍 方法二:使用 conda 安装

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

💡 需要提前配置好 Conda 环境
🖼️

pytorch_logo

🧰 方法三:从源代码编译

git clone https://github.com/pytorch/pytorch.git
cd pytorch
git checkout release/2.0.0

🔧 需要具备编译环境与依赖项
📌 查看官方编译文档

学习资源推荐

安装验证

安装完成后,通过以下代码验证是否成功:

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

✅ 若输出版本号及 True 表示安装成功
⚠️ 遇到问题请查看 常见错误排查