ModuleNotFoundError: No module named 'fused_layer_norm_cuda'
ModuleNotFoundError: No module named ‘fused_layer_norm_cuda’是apex安装出了问题安装apex步骤如下,git clone https://github.com/NVIDIA/apex.gitcd apexpython setup.py install --cpp_ext --cuda_ext参考链接:https://gith...
文章共49字 · 阅读需要大约1分钟
一键AI生成摘要,助你高效阅读
问答
·
ModuleNotFoundError: No module named ‘fused_layer_norm_cuda’
是apex安装出了问题
安装apex步骤如下,
git clone https://github.com/NVIDIA/apex.git
cd apex
python setup.py install --cpp_ext --cuda_ext
参考链接:https://github.com/NVIDIA/apex/issues/156
更多推荐
已为社区贡献2条内容
所有评论(0)