tensorflow2.5.0：tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution

问题描述

今天尝试在pycharm中运行建议的卷积神经网络，最开始使用cpu运行，但是无奈嫌弃运行速度较慢，于是在更新tensorflow-gpu后，尝试切换至gpu运行，但是程序在运行时有如下报错信息

对于这个问题，我查了很多相关的帖子，其中常见的解释无非两种：
1.cuda和cudnn版本不符
2.显存不够

对于这两种解释，我个人倾向于第二种显存不够的问题。于是参考各种帖子的对策：在代码头部添加几行代码，这里我给两种普遍的答案，各位也可以搜索得到

可以得知，这些方法的本质无非是控制显存的使用（百分比）。
我使用的是第二种v1的方法，针对于tensorflow2.0。但是在运行的过程中会提示没有compat，这就非常的疑惑，难道2.0和2.5有一些区别吗？如果有了解的大佬欢迎在评论区给予解释。（个人显卡是1060,6g显存还是有些不足了）

无奈只能切换至cpu缓慢 运行，这里给出代码，可以方便切换至cpu运行

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

欢迎各位留言讨论，如果有了解的大佬欢迎在评论区指出