From 87ffdd2d500674c28b95abec7cb9b2f7eb0bba91 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 3 May 2017 10:14:21 -0800 Subject: Internal change Change: 154978617 --- tools/tf_env_collect.sh | 108 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 108 insertions(+) create mode 100644 tools/tf_env_collect.sh (limited to 'tools') diff --git a/tools/tf_env_collect.sh b/tools/tf_env_collect.sh new file mode 100644 index 0000000000..71b17f4b7b --- /dev/null +++ b/tools/tf_env_collect.sh @@ -0,0 +1,108 @@ +#!/usr/bin/env bash +# Copyright 2017 The TensorFlow Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ============================================================================== + +set -u # Check for undefined variables + +echo "Collecting system information..." + +OUTPUT_FILE=tf_env.txt + +echo >> $OUTPUT_FILE +echo "== cat /etc/issue ===============================================" >> $OUTPUT_FILE +uname -a >> $OUTPUT_FILE +uname=`uname -s` +if [ "$(uname)" == "Darwin" ]; then + echo Mac OS X `sw_vers -productVersion` >> $OUTPUT_FILE +elif [ "$(uname)" == "Linux" ]; then + cat /etc/*release | grep VERSION >> $OUTPUT_FILE +fi + + +echo >> $OUTPUT_FILE +echo '== are we in docker =============================================' >> $OUTPUT_FILE +num=`cat /proc/1/cgroup | grep docker | wc -l`; +if [ $num -ge 1 ]; then + echo "Yes" >> $OUTPUT_FILE +else + echo "No" >> $OUTPUT_FILE +fi + +echo >> $OUTPUT_FILE +echo '== compiler =====================================================' >> $OUTPUT_FILE +c++ --version &>> $OUTPUT_FILE + +echo >> $OUTPUT_FILE +echo '== uname -a =====================================================' >> $OUTPUT_FILE +uname -a >> $OUTPUT_FILE + +echo >> $OUTPUT_FILE +echo '== check pips ===================================================' >> $OUTPUT_FILE +pip list 2>&1 | grep "proto\|numpy\|tensorflow" &>> $OUTPUT_FILE + + +echo >> $OUTPUT_FILE +echo '== check for virtualenv =========================================' >> $OUTPUT_FILE +python -c "import sys;print(hasattr(sys, \"real_prefix\"))" >> $OUTPUT_FILE + +echo >> $OUTPUT_FILE +echo '== tensorflow import ============================================' >> $OUTPUT_FILE +cat < /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.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 &>> ${OUTPUT_FILE} + +DEBUG_LD=libs python -c "import tensorflow" 2>>${OUTPUT_FILE} > /tmp/loadedlibs +grep libcudnn.so /tmp/loadedlibs >> $OUTPUT_FILE + +echo >> $OUTPUT_FILE +echo '== env ==========================================================' >> $OUTPUT_FILE +if [ -z ${LD_LIBRARY_PATH+x} ]; then + echo "LD_LIBRARY_PATH is unset" >> $OUTPUT_FILE; +else + echo LD_LIBRARY_PATH ${LD_LIBRARY_PATH} >> $OUTPUT_FILE; +fi +if [ -z ${DYLD_LIBRARY_PATH+x} ]; then + echo "DYLD_LIBRARY_PATH is unset" >> $OUTPUT_FILE; +else + echo DYLD_LIBRARY_PATH ${DYLD_LIBRARY_PATH} >> $OUTPUT_FILE; +fi + + +echo >> $OUTPUT_FILE >> $OUTPUT_FILE +echo '== nvidia-smi ===================================================' >> $OUTPUT_FILE +nvidia-smi &>> $OUTPUT_FILE + +echo >> $OUTPUT_FILE + +echo '== cuda libs ===================================================' >> $OUTPUT_FILE +find /usr/local -type f -name 'libcudart*' 2>/dev/null | grep cuda | grep -v "\\.cache" >> ${OUTPUT_FILE} +find /usr/local -type f -name 'libudnn*' 2>/dev/null | grep cuda | grep -v "\\.cache" >> ${OUTPUT_FILE} + +# Remove any words with google. +mv $OUTPUT_FILE old-$OUTPUT_FILE +grep -v -i google old-${OUTPUT_FILE} > $OUTPUT_FILE + +echo "Wrote environment to ${OUTPUT_FILE}. You can review the contents of that file." +echo "and use it to populate the fields in the github issue template." +echo +echo "cat ${OUTPUT_FILE}" +echo + -- cgit v1.2.3