进入安装目录,执行如下命令获取PyTorch插件的whl包。 wget https://gitee.com/ascend/pytorch/releases/download/v6.0.rc1-pytorch1.11.0/torch_npu-1.11.0.post11-cp38-cp38-linux_aarch64.whl PyTorch插件安装包版本需与Python版本一一对应,例如Python版本为
到处挖坑蒋玉成 踏平坎坷成大道,斗罢艰险又出发 这里说的支持指的不是官方版原生支持某一种特定的后端,而是官方版原生支持挂载插件,这样就不需要自己编译torch了。使用的方法是先安装官方CPU版torch,然后再安装torch-npu,之后import torch-npu就可以用了。 如何看待pytorch2.1,原生支持华为昇腾NPU? 发布于 2024-...