Internal Error:Blas GEMM launch failed解决方法

waiting不是违停

于 2021-07-09 08:58:34 发布

阅读量4.5k

点赞数 2

分类专栏： Python人工智能

本文链接：https://blog.csdn.net/weixin_44737922/article/details/118596152

版权

Python人工智能专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种解决TensorFlow2.x环境下GPU显存不足的方法。通过限制GPU使用率及设置显存增长策略，有效避免了因显存溢出导致的程序运行失败。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考文章链接
该问题是因为GPU显存不足
在要运行的代码前加上这段，问题便解决了（环境为TensorFlow2.x）

os.environ["CUDA_VISIBLE_DEVICES"] = '0'   #指定第一块GPU可用
config = tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.8  # 程序最多只能占用指定gpu50%的显存
config.gpu_options.allow_growth = True      #程序按需申请内存
sess = tf.compat.v1.Session(config = config)