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

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

前言

本文是我在用数据集验证DIEN模型时出现的bug-----InternalError: Blas xGEMM launch failed : a.shape=[1,4992,8], b.shape=[1,8,100], m=4992, n=100, k=8 [Op:MatMul]
这个问题可能是GPU内存不足，也可能是DUDA和cudnn不匹配。
以下是GPU内存不足的解决办法。

提示：以下是本篇文章正文内容，下面案例可供参考

一、解决办法

代码如下（示例）：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'   #指定第一块GPU可用
config = tf.compat.v1.ConfigProto()
# 程序最多只能占