您当前的位置: 首页 >  tensorflow

耐心的小黑

暂无认证

  • 3浏览

    0关注

    323博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

tensorflow2.x踩坑记录1:运行tf2.x出现如下错误:Blas GEMM launch failed或者Failed to get convolution algorithm

耐心的小黑 发布时间:2020-02-12 09:52:36 ,浏览量:3

我的第一篇CSDN博客 一、运行新版的tf2.0或者tf2.1出现如下错误:
Blas GEMM launch failed : a.shape=(300, 1), b.shape=(1, 10), m=300, n=10, k=1
  • 经过查找相关资料和自己试验,发现这可能和GPU的内存分配有关系;因为默认情况下在代码中使用GPU时,有把内存占满的趋势;即使有时候计算的数据量并不足以占用整个GPU。
  • 所以,我们的思路之一就是对GPU的使用模式进行设置,如下面的代码所示,我们将GPU设置为memory_growth模式,它的意思是此时我们需要多少GPU内存就使用多少,不会过多占用。
##列出你所有的物理GPU
gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config
关注
打赏
1640088279
查看更多评论
立即登录/注册

微信扫码登录

0.0376s