这可能是第一篇比较全的ERNIE踩坑笔记——Centos下玩转ERNIE

本文详细记录了在Centos系统中使用Anaconda环境搭建、安装PaddlePaddle 1.3.0、下载ERNIE源码并进行训练的过程,包括配置实验任务、解决CUDA环境问题以及训练模型时的注意事项。作者分享了在实践ERNIE模型时遇到的问题和解决方案,旨在帮助读者避免相同陷阱。
摘要由CSDN通过智能技术生成

最近要做一个中文文本匹配的任务,看到百度刚刚发布了sota的模型(中文任务全面超越BERT:百度正式发布NLP预训练模型ERNIE) ,所以想拿来用一下,但是相对于大火的google发布的Bert,实践ERNIE模型的博客非常少,于是自己跑通模型后写了这篇博客记录一下遇到的坑,如果对小伙伴们有帮助那便是极好的。

因为论文是基于百度的开源框架paddlepaddle,可以说相比较深度学习大火的tensorflow和pytorch,这方面的资料非常少,可以参考的有官方文档和Github,这里也是给出我在自己环境中遇到的一些问题和解决方案。

文章会同步到我的微信公众号编程与算法之美,公众号底部菜单栏有我的交流群,欢迎大家关注。

写作不易,转载请注明出处。

目录

系统环境

anaconda下创建环境

安装paddlepaddle

下载ERNIE源码

配置实验任务

训练模型

实验结果

后记


系统环境

# linux 版本
#这种方法只适合Redhat系的Linux,我目前尝试只有这个可以查看Centos系统版本
cat /etc/redhat-release 
-> CentOS Linux release 7.4.1708 (Core)

# cuda 版本 
cat /usr/local/cuda/version.txt
-> CUDA Version 9.2.148

cudnn 版本 
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
-> #define CUDNN_MAJOR 7
-> #define CUDNN_MINOR 3
-> #de
  • 4
    点赞
  • 29
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值