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 安装详解

PyTorch_CPU_Installation