PyTorch 是一个广泛使用的深度学习框架,以下是不同环境下的安装方法:
💻 Linux 系统安装
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
🐧 macOS 系统安装
pip install torch torchvision torchaudio
🖥️ Windows 系统安装
pip install torch torchvision torchaudio
📌 选择版本与环境
- CUDA 支持:需先安装 NVIDIA 驱动,建议使用
cu118
版本(对应 CUDA 11.8) - Conda 用户:可通过
conda install pytorch torchvision torchaudio -c pytorch
安装 - 离线安装:参考 PyTorch 官方文档 获取包文件
📚 扩展学习
如需了解 PyTorch 的应用场景,可访问 PyTorch 应用案例 查看详细教程。