当前位置:   article > 正文

【Linux】GPU运行BERT环境搭建(CUDA9.0,cudnn7.0,tensorflow-gpu1.11.0)_bert 支持的cuda

bert 支持的cuda


菜鸟选手辛辛苦苦终于做好了环境,以防以后还需要搭建这样的环境,写个备忘

  • 目的:为了在服务器Linux系统上运行基于原始tensorflow版本的bert,为了把代码挪到GPU上跑

BERT版本

GitHub上原始代码,因为它主要用的是tensorflow1.11.0,咱们也保持一致,就用这个版本,对应这个版本,需要的CUDA是9.0,cudnn是7.0

先装CUDA9.0

找到官网去下载安装包(https://developer.nvidia.com/cuda-90-download-archive),对应系统型号,有三种版本,一种是.run,一种是体积很小的包&#

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/知新_RL/article/detail/640136
推荐阅读
相关标签
  

闽ICP备14008679号