Tensorflow使用中最常见的错误——distributed 显存错误

在Tensorlfow课程答疑过程中,一个很常见的错误就是显存分配错误。

常常会包含如下的错误提示:

Interal Error:Blas GEMM launch failed.

这个错误是GPU显存分配错误,也就是说显存不足造成的。

这里有两种情况,

第一,你的数据很小,却产生这个报错。

这是由于 Tensorflow 默认申请可使用的全部显存,当tensorflow程序运行会话却没有关闭会话释放资源的时候,就会出现此错误,尤其是使用 jupyter notebook 的时候。可使用 nvidia-smi 命令查看NVIDIA显卡的显存占用情况。

解决这个问题的办法是:设置显存按需申请。

在代码(注意这里的适用版本为tensorlfow 2.0及以上版本)前加上:

gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)

重启程序或kernel运行。

第二,显存太小,确实不够了。

这时,需要减小 batch_size, 减小输入图片大小,甚至使用小的模型, 另外,对于图片的加载推荐使用tf.data.Datatset或生成器。

  • 0
    点赞
  • 0
    收藏
  • 打赏
    打赏
  • 0
    评论

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

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
©️2022 CSDN 皮肤主题:大白 设计师:CSDN官方博客 返回首页
评论

打赏作者

日月光华老师

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值