<深度學習>Tensorflow遇到的坑之一

AttributeError: module 'tensorflow' has no attribute 'random_normal'

AttributeError: module 'tensorflow' has no attribute 'Session'

 

查了很多，發現是tensorflow2.0不支持session了，之前試了

sess = tf.compat.v1.Session()

這句無效。

with tf.compat.v1.Session() as ses:

　　# Build a graph.
　　a = tf.constant(5.0)
　　b = tf.constant(6.0)
　　c = a * b

　　# Evaluate the tensor `c`.
　　print(ses.run(c))

 

這種寫法可以解決，但是不兼容，我換個代碼就不行了。。

 

然后在博客的回復里看到一個，可以解決這個問題：

import os
import tensorflow as tf
import matplotlib.pyplot as plt
import tensorflow.compat.v1 as tf
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

tf.disable_v2_behavior()
a = tf.random_([2, 30])
sess = tf.Session()
out = sess.run(a)
x, y = out
plt.scatter(x, y)
plt.show()

導入的時候直接導入tensorflow1的包