mirror of
https://github.com/zebrajr/tensorflow.git
synced 2025-12-06 12:20:11 +01:00
also update the tf_env_collect script to work with 2.0
This commit is contained in:
parent
60e892e4bc
commit
ad10339dad
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user