diff --git a/checks/pip_check.sh b/checks/pip_check.sh
index 20b745c..26fb541 100644
--- a/checks/pip_check.sh
+++ b/checks/pip_check.sh
@@ -25,6 +25,11 @@ for r in ${REQUIRED[@]}; do
       if [ $r = ${q} ]; then
         present=1
       fi
+      if [ $r = 'tensorflow' ]; then
+        if [ 'tensorflow-cpu' = $q ]; then
+          present=1
+        fi
+      fi
     fi
   done
   if (($present == 0))