Skip to content

Commit

Permalink
Internal change
Browse files Browse the repository at this point in the history
Change: 154978617
  • Loading branch information
tensorflower-gardener committed May 3, 2017
1 parent 427081c commit 87ffdd2
Show file tree
Hide file tree
Showing 2 changed files with 121 additions and 0 deletions.
13 changes: 13 additions & 0 deletions third_party/ortools.BUILD
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# Google's software suite for combinatorial optimization

licenses(["notice"]) # Apache2 license

exports_files(["LICENSE-2.0.txt"])

native.cc_library(
name = "linear_solver_glop",
deps = [
"@ortools_archive//linear_solver:linear_solver_glop",
],
visibility = ["//visibility:public"],
)
108 changes: 108 additions & 0 deletions tools/tf_env_collect.sh
Original file line number Diff line number Diff line change
@@ -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:https://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 <<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.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

0 comments on commit 87ffdd2

Please sign in to comment.