欢迎使用 PyTorch 安装教程!无论您是新手还是有经验的开发者,本指南都将帮助您顺利安装 PyTorch。📌
安装前的准备 🛠️
安装方法 📚
使用 Conda 安装 🐍
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
使用 Pip 安装 📦
pip3 install torch torchvision torchaudio
验证安装 ✅
运行以下代码检查是否安装成功:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
若输出版本号且显示 True
,则说明 GPU 支持已启用!🔥
常见问题 ❓
✨ 安装完成后,可前往 PyTorch 基础教程 开始学习深度学习模型构建!