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

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐