华为晟腾计算芯片 —— AI国产化 —— Ascend / pytorch —— 支持NPU的pytorch下载地址

地址：

https://gitee.com/ascend/pytorch/releases

上面的这个下载地址可能出现问题，不能有新的版本发布了，于是有了下面使用华为软件库直接在线安装。

首先配置你的pip下载地址：

http://repo.huaweicloud.com/repository/pypi/simple

配置好后pip直接安装：

pip install torch-npu

或者

pip install torch-npu -i http://repo.huaweicloud.com/repository/pypi/simple

运行编写的测试代码：

import torch_npu  # 导入torch-npu
 
# 尝试创建两个张量并放到NPU上做矩阵乘法
x = torch.randn(20000, 20000).npu()  # .npu()方法将张量移至NPU
y = torch.randn(20000, 20000).npu()
for _ in range(1111111111111111):
    z = x.mm(y)  # 矩阵乘法

查看npu运行情况：

watch -n 1 npu-smi info

成功在晟腾910B上运行pytorch。