转换google 官方 bert向量为Pytorch可用版本-白红宇的个人博客

发布日期：2021-05-04 21:01:55 浏览次数：13 分类：技术文章

本文共 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" .

上一篇：python中快速删除重复元素

下一篇：pytorch 中加载 bert 模型

关于作者

喝酒易醉，品茶养心，人生如梦，品茶悟道，何以解忧？唯有杜康！

-- 愿君每日到此一游！

从零构建通讯器--5.2三次握手，telnet,wireshark 2019-03-01

如何判断两个浮点数是否相等？ 2019-03-01

2017ccpc杭州 E. Master of Subgraph（点分治 + 树dp + bitset） 2019-03-01

2021牛客寒假算法基础集训营3 2019-03-01

苹果进军搜索，背后藏着什么“阳谋”？ 2019-03-01

egg：如何在控制器中拿到前端传的参数 2019-03-01

MVC之修改 2019-03-01

struct 模块 2019-03-01

python之面向对象编程 2019-03-01

python之集合类型内置方法 2019-03-01

编程与编程语言分类 2019-03-01

【 UVA - 572 】 Oil Deposits （DFS水题） 2019-03-01

约瑟夫环问题 2019-03-01

CF #716 (Div. 2) B. AND 0, Sum Big（思维+数学） 2019-03-01

Java 設計模式 - 建造者模式 2019-03-01

ES6 JavaScript 重新認識 Promise 2019-03-01

2020-07-16：如何获得一个链表的倒数第n个元素？ 2019-03-01

2021-01-21：java中，HashMap的读流程是什么？ 2019-03-01

白红宇的个人博客 - 记录点点滴滴的事 - 您是第 453435414 位访客