解决“Errors may have originated from an input operation. Input Source operations connected to node“

项目场景:

提示:这里简述项目相关背景:
Tensorflow-gpu训练卷积时报出错误,虽然可以启动但无法训练
环境
pycharm
+tensorflow-gpu2.4.0
+python3.8
+cuda11.1

问题描述

提示:这里描述项目中遇到的问题:

报错
很生气
1
1
1

Errors may have originated from an input operation.
Input Source operations connected to node Conv2D:

原因分析:

tensorflow2.X与tensorflow1.X不同

解决方案:

添加以下三行代码即可解决问题

config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.compat.v1.Session(config=config)

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
xiaoxingxing的头像xiaoxingxing管理团队
上一篇 2022年3月31日
下一篇 2022年4月1日

相关推荐