PyTorch 是深度学习领域广泛使用的框架,以下是几种常见安装方式:
1. 从源码安装 🧰
git clone https://github.com/pytorch/pytorch.git
cd pytorch
⚠️ 需要提前安装 CMake 和其他依赖项
2. 使用 Conda 环境 🐍
conda create -n pytorch_env python=3.9
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit -c pytorch
3. 使用 Pip 安装 📦
pip install torch torchvision torchaudio
📌 请确保已配置好 虚拟环境 管理
4. 验证安装 ✅
import torch
print(torch.__version__)
print(torch.cuda.is_available())
📌 扩展阅读:PyTorch 快速入门