Python 中导入 PyTorch 时使用 import torch 而非 import pytorch

在 Python 中导入 PyTorch 时使用 import torch 而非 import pytorch，这是 PyTorch 官方设计的历史惯例。以下是详细解释：

---

一、包名设计原因

1. 历史命名传统：

   • PyTorch 的前身是 Torch（基于 Lua 的深度学习框架），其 Python 版本延续了 torch 作为核心包名。

   • 开发团队选择保持名称简洁性，避免与早期项目混淆。

2. 避免名称冲突：

   • 若包名为 pytorch，可能与某些第三方库（如 pytorch-lightning）产生命名空间冲突。

   • 所有 PyTorch 生态组件（如 torchvision、torchaudio）均以 torch 为统一前缀。

3. 官方文档与生态统一性：

   • 所有 PyTorch 官方文档、示例代码均使用 import torch，确保开发者体验一致。

---

二、为什么安装命令是 pip install torch？

• PyPI 注册名称：PyTorch 在 PyPI（Python 包索引）上注册的官方包名是 torch，而非 pytorch。

• 验证方法：

  # 查看已安装的 PyTorch 包名
  pip list | grep torch

  输出示例：

  torch           2.2.1
  torchvision     0.17.1

---

三、常见误区

1. 为何存在 pytorch 这个包？

• 非官方包：PyPI 上的 pytorch 是一个非官方、废弃的占位包（由第三方用户注册），与 PyTorch 官方无关。

• 安装陷阱：如果误执行 pip install pytorch，会安装一个空包或旧版本，导致代码运行失败。

2. 正确安装 PyTorch 的方式

通过 PyTorch 官网提供的命令安装（访问官网选择配置生成命令）：

# CPU 版本
pip install torch --index-url https://download.pytorch.org/whl/cpu

# CUDA 12.x GPU 版本
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

---

四、验证 PyTorch 安装

import torch

# 输出版本号（如 2.2.1）
print(torch.__version__)  

# 检查 GPU 是否可用
print(torch.cuda.is_available())  # True 表示 GPU 支持已启用

---

五、总结

关键点	说明
导入语句	使用 import torch，这是 PyTorch 的官方约定
安装包名	通过 pip install torch 安装，而非 pytorch
历史原因	继承自 Torch 框架，保持生态一致性
避免非官方包	不要安装 pytorch 包，它是无效的占位符

---

通过以上解释，可以明确 PyTorch 的包名是 torch，这一设计是官方有意为之的规范。