[機器學習]AttributeError: module 'tensorflow' has no attribute 'ConfigProto' 報錯解決方法


在代碼:

   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) 

修改后代碼:

 1 #!/usr/bin/env python
 2 # -*- coding: utf-8 -*-
 3 # @File  : TensorFlow入門操作.py
 4 # @Author: 趙路倉
 5 # @Date  : 2020/3/26
 6 # @Desc  :
 7 # @Contact : 398333404@qq.com 
 8 
 9 import tensorflow as tf
10 import os
11 os.environ["CUDA_VISIBLE_DEVICES"]="0"
12 
13 tf.compat.v1.disable_eager_execution()
14 hello=tf.constant('Hello,TensorFlow')
15 config=tf.compat.v1.ConfigProto()
16 config.gpu_options.per_process_gpu_memory_fraction = 0.9
17 sess=tf.compat.v1.Session(config=config)
18 print(sess.run(hello))

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM