diff --git a/requirements-node.txt b/requirements-node.txt
index 179f2e5..0fab3fa 100644
--- a/requirements-node.txt
+++ b/requirements-node.txt
@@ -4,4 +4,6 @@ pillow
 portpicker
 setuptools
 simplejson
-tensorflow
+tensorflow_cpu
+#tensorflow
+#tensorflow_gpu