关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>

New Issue

#4983 需要NPU的torch2.1.0支持

Open

created 4 months ago by zzc0208 · 6 comments

问题描述

需要NPU的torch2.1.0支持

相关环境（GPU/NPU）

NPU

相关集群（启智/智算）

智算

日志说明或问题截图

需要NPU的torch2.1.0支持
官方仓库，需要编译安装，需要root权限，本人搞不定：
https://gitee.com/ascend/pytorch

期望的解决方案或建议

需要NPU的torch2.1.0支持
官方仓库，需要编译安装，需要root权限，本人搞不定：
https://gitee.com/ascend/pytorch

tanglj added the

need review

label 3 months ago

tanglj added this to the V20240129 milestone 3 months ago

liuzx was assigned by tanglj 3 months ago

智算npu调试已支持torch2.1

liuzx added the

test

label 2 months ago

智算NPU调试已添加镜像 mindtorch0.2_mindspore2.2.1_torchnpu2.1.0_cann7.0rc1 ，支持torch2.1。

@liuzx 启智NPU调试，镜像pytorch2.1_mindspore2.2_cann7，未完整支持torch2.1，代码运行报错。

验证代码如下（来源：https://gitee.com/ascend/pytorch）
import torch
import torch_npu

x = torch.randn(2, 2).npu()
y = torch.randn(2, 2).npu()
z = x.mm(y)

print(z)

45 KiB

wangj removed the

test

label 1 month ago

wangj modified the milestone from V20240129 to V20240402 1 month ago

启智npu调试镜像torch2.1已更新并上线生产，智算npu调试也已上线torch2.1

liuzx added the

test

label 1 month ago

启智npu调试镜像pytorch2.1_cann7，验证结果：
1.在terminal跑验证代码，有大段warning，最后输出了计算结果。
2.点击Launcher——Notebook——Pytorch2.1创建的notebook，no kernel

wangj removed the

test

label 1 month ago

wangj modified the milestone from V20240402 to V20240423 3 weeks ago

需要华为技术支持，暂时挂起

wangj added the

wait

label 6 days ago

wangj modified the milestone from V20240423 to V20240520 4 days ago

Sign in to join this conversation.

wait

V20240520

3 Participants

Notifications

Due Date

No due date set.

Dependencies

This issue currently doesn't have any dependencies.