Trouble-Shooting TensorFlow

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