Skip to content

Instantly share code, notes, and snippets.

@PatWie

PatWie/tf_info.sh

Created January 11, 2019 09:30
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save PatWie/c04f4d6a61ac201b8afa2c59fe9ba586 to your computer and use it in GitHub Desktop.
Save PatWie/c04f4d6a61ac201b8afa2c59fe9ba586 to your computer and use it in GitHub Desktop.
Download ZIP
Trouble-Shooting TensorFlow
Raw
tf_info.sh
# get os info
uname -as
# get compiler info
g++ --version
# check python library versions
pip list 2>&1 | grep "proto\|numpy\|tensorflow"
# get TensorFlow information
cat <<EOF > /tmp/check_tf.py
import tensorflow as tf;
print("path = %s" % tf.__file__)
print("tf.GIT_VERSION = %s" % tf.GIT_VERSION)
print("tf.VERSION = %s" % tf.VERSION)
print("tf.COMPILER_VERSION = %s" % tf.GIT_VERSION)
with tf.Session() as sess:
print("Sanity check: %r" % sess.run(tf.constant([1,2,3])[:1]))
EOF
python /tmp/check_tf.py
# get cuda information when necessary
nvidia-smi
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment