当前位置: 首页 > news >正文

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

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

地址:

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


image


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




首先配置你的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





image


image