已关闭
torch_npu_install.sh 用 pip3 安装却用 $PYTHON_CMD 验证,解释器不一致时会装到错误环境 #41
崇理战队创建于  8月13日关闭于  24 天前
崇理战队
8月13日 创建

torch_npu_install.sh 在第 1 步检测并记录了当前 Python 解释器:

PYTHON_CMD=""
if command -v python3 ...; then PYTHON_CMD="python3"
elif command -v python ...; then PYTHON_CMD="python"

但第 4/5 步实际安装时用的是硬编码的 pip3

pip3 install "$TORCH_WHL" -U
...
pip3 install "$NPU_WHL" -U

最后验证又回到 $PYTHON_CMD

$PYTHON_CMD -c "import torch; import torch_npu; ..."

当环境里 pythonpython3 指向不同解释器时(典型如:conda/venv 激活后只接管了 python,或系统 python3 与用户 python 并存),pip3 绑定的解释器可能并不是 $PYTHON_CMD——包装进了另一个环境,验证步骤随即报 ModuleNotFoundError,而且错误信息指向 import 失败而非安装目标错误,很容易误导排查。

建议统一改为 $PYTHON_CMD -m pip install ...,保证安装与验证使用同一解释器。

另外一个小问题:脚本顶部 TORCH_NPU_VERSION="7.3.0" 只用于 echo 展示,实际下载的 wheel 版本是 torch_npu-${TORCH_VERSION}.post2(2.7.1.post2),两个"版本"口径不同容易让维护者误解,建议注释说明或去掉该变量。

likedislike
kong0808成员
29 天前 评论:

您好!感谢关注,我们将尽快分析和处理!

likedislike
Kkong0808成员
29 天前 将 li-yuanjie-da 设为负责人
Lli-yuanjie-da成员
28 天前 关联了pull request:modify torch_npu_install.sh
li-yuanjie-da成员
27 天前 评论:

@jiangzeyu-2026

您好,感谢反馈。
脚本虽然检测并记录了Python解释器,但安装阶段使用了独立的pip3命令,无法保证安装和验证使用同一个解释器。我们统一改为"${PYTHON_CMD}" -m pip,并增加pip可用性检查。
关于版本变量,7.3.0表示torch_npu发行版本,wheel中的2.7.1.post2是Python包版本,两者口径不同。当前命名和展示确实容易产生歧义,已通过变量重命名和注释明确两者关系。

likedislike
li-yuanjie-da成员
25 天前 评论:

@jiangzeyu-2026

该issues即将关闭。

likedislike
Lli-yuanjie-da成员
24 天前 issue状态由 进行中 改变为 已完成
Lli-yuanjie-da成员
24 天前 关闭了 issue
CANN-robotCANN-robot成员
24 天前 添加了label:resolved