bert 内存不足

tzxcd3kk  于 3个月前  发布在  其他
关注(0)|答案(1)|浏览(41)

当BERT模型加载时,出现了一些问题:

  • tensorflow.python.framework.errors_impl.InternalError: CUDA运行时隐式初始化失败。状态:内存不足*

但是,GPU内存还剩下20G。运行环境如下:
CentOS 7
tensorflow-gpu==1.11.0
cuda==9.0.176
cudnn=7.0.5

相关问题