AttributeError: module 'tensorflow' has no attribute 'ConfigProto'
时间: 2023-09-24 18:06:21 浏览: 173
问题解决:module 'tensorflow.compat.v1' has no attribute '
5星 · 资源好评率100%
这个错误可能是因为你使用的 TensorFlow 版本与代码中使用的不兼容。`ConfigProto` 是 TensorFlow 中的一个配置类,如果你使用的是较旧的 TensorFlow 版本,可能不支持该属性。
你可以尝试更新 TensorFlow 版本,或者更改代码中的部分以适应你的 TensorFlow 版本。如果你使用的是 TensorFlow 1.x 版本,可以使用以下代码:
```
import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)
```
如果你使用的是 TensorFlow 2.x 版本,可以使用以下代码:
```
import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus:
try:
for gpu in gpus:
tf.config.experimental.set_memory_growth(gpu, True)
except RuntimeError as e:
print(e)
```
阅读全文