[机器学习]AttributeError: module 'tensorflow' has no attribute 'ConfigProto' 报错解决方法

最新推荐文章于 2024-12-04 14:31:46 发布

原创最新推荐文章于 2024-12-04 14:31:46 发布 · 1.2w 阅读

29 ·

CC 4.0 BY-SA版权

本文介绍在TensorFlow2.0环境下，如何解决因版本更新导致的代码不兼容问题，具体展示了如何修改ConfigProto和Session配置，以确保GPU内存分配正确并成功运行示例代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在代码：

　　 config=tf.ConfigProto()

　　 sess=tf.compat.v1.Session(config=config)

执行过程中会报错

　　config=tf.ConfigProto()
AttributeError: module 'tensorflow' has no attribute 'ConfigProto'

问题原因：因为是tensorflow 2.0版本与1.0的用法不兼容

　　解决办法：

修改为 config=tf.compat.v1.ConfigProto() 和 sess=tf.compat.v1.Session(config=config)

修改后代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @File  : TensorFlow入门操作.py
# @Author: 赵路仓
# @Date  : 2020/3/26
# @Desc  :
# @Contact : 398333404@qq.com 

import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"]="0"

tf.compat.v1.disable_eager_execution()
hello=tf.constant('Hello,TensorFlow')
config=tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.9
sess=tf.compat.v1.Session(config=config)
print(sess.run(hello))