also update the tf_env_collect script to work with 2.0

This commit is contained in:
serv-inc 2019-04-05 14:37:46 +02:00
parent 60e892e4bc
commit ad10339dad

View File

@ -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