apex的安装和使用 fused_weight_gradient_mlp_cuda module is not found
收藏回复举报
apex的安装和使用 fused_weight_gradient_mlp_cuda module is not found
t('forum.solved') 已解决
发表于2024-07-17 20:22:46
0 查看

按照(可选)安装APEX模块-安装步骤-配置与安装-Ascend Extension for PyTorch6.0.RC1开发文档-昇腾社区 (hiascend.com)这个流程成功安装了apex

cke_1198.png

但是使用python命令行import时报错:

cke_4422.png

进行模型推理时,相应地会报错:

cke_12080.png

按照提示,在pip安装的时候加上--cpp_ext --cuda_ext不能解决问题,应该不是得装cuda环境吧?毕竟这是昇腾npu

我要发帖子