错误：InternalError: Blas xGEMM launch failed 的解决办法

最新推荐文章于 2024-07-31 18:38:45 发布

吃午饭等晚饭777

最新推荐文章于 2024-07-31 18:38:45 发布

阅读量655

点赞数

文章标签：运维 Powered by 金山文档

原文链接：https://blog.csdn.net/weixin_43560286/article/details/123734879?spm=1001.2101.3001.6650.2&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-2-123734879-blog-91381449.235%5Ev27%5Epc_relevant_3mothn_strategy_recovery&depth_1-utm_

版权

这个问题可能是GPU内存不足，也可能是DUDA和cudnn不匹配。

以下是GPU内存不足的解决办法。

解决办法

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