PyTorch 是一个广泛使用的深度学习框架,支持 CPU 和 GPU 运算。以下是在 CPU 上安装 PyTorch 的详细步骤:
安装方法📋
1. 通过 pip 安装
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
📌 注意:确保已安装 Python 3.8+ 和 pip。
2. 通过 Conda 安装
conda install pytorch torchvision torchaudio cpuonly -c pytorch
📌 注意:需使用 Anaconda 环境。
3. 验证安装
运行以下代码检查是否成功:
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # 应显示 False(CPU 模式)
常见问题⚠️
- 问题:安装后无法导入 PyTorch?
- 解决:检查 Python 环境路径是否正确,或尝试重新安装。
- 问题:需要额外依赖?
- 解决:安装 libomp 库以解决某些版本的兼容性问题。
扩展阅读📘
如需了解 PyTorch 的更多用法,可访问 PyTorch 官方文档 或 PyTorch CPU 安装详解。