Try both python and python3
Used `configure` script for reference
This commit is contained in:
parent
fa3884883d
commit
da70b83f35
@ -19,6 +19,7 @@ set -u # Check for undefined variables
|
||||
echo "Collecting system information..."
|
||||
|
||||
OUTPUT_FILE=tf_env.txt
|
||||
python_bin_path=$(which python || which python3 || true)
|
||||
|
||||
{
|
||||
echo
|
||||
@ -55,7 +56,7 @@ OUTPUT_FILE=tf_env.txt
|
||||
|
||||
echo
|
||||
echo '== check for virtualenv ========================================='
|
||||
python -c "import sys;print(hasattr(sys, \"real_prefix\"))"
|
||||
${python_bin_path} -c "import sys;print(hasattr(sys, \"real_prefix\"))"
|
||||
|
||||
echo
|
||||
echo '== tensorflow import ============================================'
|
||||
@ -69,9 +70,9 @@ 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 2>&1 >> ${OUTPUT_FILE}
|
||||
${python_bin_path} /tmp/check_tf.py 2>&1 >> ${OUTPUT_FILE}
|
||||
|
||||
DEBUG_LD=libs python -c "import tensorflow" 2>>${OUTPUT_FILE} > /tmp/loadedlibs
|
||||
DEBUG_LD=libs ${python_bin_path} -c "import tensorflow" 2>>${OUTPUT_FILE} > /tmp/loadedlibs
|
||||
|
||||
{
|
||||
grep libcudnn.so /tmp/loadedlibs
|
||||
|
Loading…
Reference in New Issue
Block a user