PyTorch 是一个广泛使用的深度学习框架,以下是几种常见安装方式:
1. 使用 pip 安装 📦
pip install torch torchvision torchaudio
⚠️ 注意:请确保已安装 Python 环境。
2. 使用 conda 安装 🐍
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch
🌟 详细配置可参考 Conda 安装文档
3. 从源码编译 🧰
git clone https://github.com/pytorch/pytorch
cd pytorch
python setup.py install
📌 需要较高技术水平,建议优先使用官方预编译版本
验证安装 ✅
运行以下代码检查是否成功:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
🚀 若输出版本号和
True
,说明安装完成
如需了解 PyTorch 基础用法,可继续阅读 PyTorch 入门教程