also update the tf_env_collect script to work with 2.0

akuroda · Apr 5, 2019 · ad10339 · ad10339
1 parent 60e892e
commit ad10339
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tools/tf_env_collect.sh b/tools/tf_env_collect.sh
@@ -74,9 +74,9 @@ python_bin_path=$(which python || which python3 || die "Cannot find Python binar
 
 cat <<EOF > /tmp/check_tf.py
 import tensorflow as tf;
-print("tf.VERSION = %s" % tf.VERSION)
-print("tf.GIT_VERSION = %s" % tf.GIT_VERSION)
-print("tf.COMPILER_VERSION = %s" % tf.COMPILER_VERSION)
+print("tf.version.VERSION = %s" % tf.version.VERSION)
+print("tf.version.GIT_VERSION = %s" % tf.version.GIT_VERSION)
+print("tf.version.COMPILER_VERSION = %s" % tf.version.COMPILER_VERSION)
 with tf.Session() as sess:
   print("Sanity check: %r" % sess.run(tf.constant([1,2,3])[:1]))
 EOF