install_pytorch

安装 PyTorch 指南

🚀 选择安装方式

1. 推荐方式 - 使用 pip 或 conda 安装
   ⚠️ 确保已安装 Python 环境
   ✅ 执行命令：

   pip install torch torchvision torchaudio
   

   或

   conda install pytorch torchvision torchaudio -c pytorch
   

2. 自定义安装 - 从源码编译（需 CUDA 支持）
   📌 详见 PyTorch官方文档 获取完整配置指南
   ⚠️ 建议先验证 GPU 环境要求

📌 安装验证

运行以下代码确认是否成功：

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

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

📷 相关资源

需要更多帮助？访问 PyTorch社区指南 获取支持！