首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >导入apex报错AttributeError: module ‘torch‘ has no attribute ‘library‘

导入apex报错AttributeError: module ‘torch‘ has no attribute ‘library‘

作者头像
云未归来
发布2025-07-22 15:04:58
发布2025-07-22 15:04:58
1390
举报

使用torch==1.11.0+cu113会报错

always have Traceback (most recent call last): File "", line 1, in File "/root/apex/apex/init.py", line 26, in from . import normalization File "/root/apex/apex/normalization/init.py", line 1, in from .fused_layer_norm import FusedLayerNorm, MixedFusedLayerNorm, FusedRMSNorm, MixedFusedRMSNorm File "/root/apex/apex/normalization/fused_layer_norm.py", line 70, in if supports_custom_op(): File "/root/apex/apex/normalization/fused_layer_norm.py", line 18, in supports_custom_op return hasattr(torch.library, "custom_op") AttributeError: module 'torch' has no attribute 'library'。

解决方法:

git clone https://github.com/NVIDIA/apex.git cd apex git tag # if not wanted version git branch -r git checkout origin/22.02-parallel-state # or another vesion python setup.py install --cuda_ext --cpp_ext pip3 install -v --no-cache-dir ./

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2025-06-30,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档