diff --git a/tools/tf_env_collect.sh b/tools/tf_env_collect.sh
index 3740ee41bb5..b7b8c25ee43 100755
--- a/tools/tf_env_collect.sh
+++ b/tools/tf_env_collect.sh
@@ -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