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
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