tensorflow报错:tensorflow.python.framework.errors_impl.InternalError: Blas GEMM launch failed :

#在tensorflow 2.0 里面,要想一个高阶迭代多次调用tf.GradientTape()时因为tape是一次性的,算完就会释放,所以要想重复调用必须设置persistent=’True‘,但是如果忘记了释放就会导致GPU被占用

w = tf.constant(1.)
x = tf.constant(2.)
with tf.GradientTape(persistent='True') as tape:
    #tape是一次性的算完就会释放,所以要想重复调用,设置persistent=’True‘,但是记得释放因为很占内存
    tape.watch([w])#跟踪参数的梯度,必须要,不然就会出现None的情况
    y = x*w  

报错:tensorflow.python.framework.errors_impl.InternalError: Blas GEMM launch failed :

解决办法,关闭GradientTape的内存占用

发布了70 篇原创文章 · 获赞 5 · 访问量 3505

猜你喜欢

转载自blog.csdn.net/qq_42647903/article/details/103906131