在代码:
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))