项目场景:
提示:这里简述项目相关背景:
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)
文章出处登录后可见!
已经登录?立即刷新