No module named 'fused_layer_norm_cuda' · Issue #161 · NVIDIA/apex · GitHub
按照以下方式安装,可解决问题
apex 安装步骤:
1. git clone https://github.com/NVIDIA/apex.git
2. cd apex
3.
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .