PyTorch 是一个流行的深度学习框架,支持多种安装方式。以下是适用于 Linux 的安装步骤:
1. 使用 pip 安装
推荐通过官方预编译包安装,简单快捷:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
✅ 适用于 Python 3.8+ 环境
⚠️ 确保已安装 CUDA 驱动(如需 GPU 加速)
2. 使用 conda 安装
如果你使用 Anaconda 环境,可通过以下命令安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c conda-forge
📦 适合科学计算环境管理
🔗 了解更多 conda 安装细节
3. 从源码编译
需较高技术门槛,适合定制化需求:
git clone https://github.com/pytorch/pytorch.git
cd pytorch
git checkout v2.4.1
python3 setup.py build --build-lib /path/to/install
🛠️ 需安装 依赖库(如 CMake、Ninja 等)
常见问题排查 🛠️
- 版本兼容性:检查 PyTorch 官方文档
- CUDA 问题:可参考 CUDA 安装指南
- 虚拟环境:建议使用
virtualenv
或conda
隔离依赖
📌 提示:安装完成后,可通过以下命令验证:
python3 -c "import torch; print(torch.__version__)"
🎉 成功安装后,你将可以开始使用 PyTorch 进行机器学习实验!