他的回复:
你好,可以看下我另一个帖子:https://bbs.huaweicloud.com/forum/thread-0243192718859779032-1-1.html cann 和 torch 可以安装,但是安装后无法正常 import torch,报错参考上述帖子。 我怀疑是因为 npu 底层驱动版本和 torch_npu 版本不兼容,参考 torch_npu 的官方 gitee,有如下说明: “ 五. 特殊声明 • 虚拟内存与单进程多卡需要在Ascend HDK 24.1.RC3以上的版本才能直接使用,其他版本不能共同使用 ” 原文来自:https://gitee.com/ascend/pytorch/releases 版本:release v7.1.0-pytorch2.6.0 notebook用户无法手动安装底层驱动,所以需要提供下 8.2.RC1 配套的镜像。