本文共 265 字，大约阅读时间需要 1 分钟。

原文： 

需要将bert预训练 模型转换成pytorch可用版本，遇到下面的错误

ModuleNotFoundError: No module named 'fused_layer_norm_cuda'

解决办法：

下载apex 并安装

# git clone https://github.com/NVIDIA/apex.git

# cd apex

# pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .

转载地址：http://bpwl.baihongyu.com/