ImportError: TFBertForSequenceClassification requires the TensorFlow library but it was not found

在用本地pycharm环境运行kaggle上面代码时弹出如标题的错误,完整报错如下:


> ImportError:  TFBertForSequenceClassification requires the TensorFlow
> library but it was not found in your environment. Checkout the
> instructions on the installation page:
> https://www.tensorflow.org/install and follow the ones that match your
> environment.

这个TFBertForSequenceClassification是transformers中的一个库。
博主尝试了各种解决方案,均无果。(关键是百度谷歌都没有过这种报错)o(╥﹏╥)o
最后解决方案为回归到本质,阅读报错信息可以发现,它提示版本不对,于是尝试从tf1.15版本变到2.0版本,依然报错。然后博主灵感一现,既然在kaggle上面可以运行,那么就找到kaggle的环境版本是多少,于是机智的博主在kaggle代码里面添加了如下代码:

import tensorflow as tf
import transformers
print(transformers.__version__)
print(tf.__version__)

在这里插入图片描述
于是,博主便对照着tf和transformers的版本一一安装,最终debug成功!(两个版本都要严格对应,博主尝试过只安装其中一个,如只装tf,不装transformers,结果依然报错)
最后附上:

  1. tensorflow版本安装方法:Linux快速搭建tensorflow-gpu 1.15.0环境(娇羞⁄(⁄ ⁄•⁄ω⁄•⁄ ⁄)⁄地献出我之前写的blog,只要把里面的第四点从1.15.0改为2.3.1即可,第五点的cudatoolkit=10.0改为cudatoolkit=10.1)
  2. transformers版本安装方法:这个很简单,直接输入以下命令即可自动卸载以前版本并安装对应版本:
pip install transformers==3.4.0

成功运行的截图:
在这里插入图片描述

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值