diff --git a/installator.sh b/installator.sh
index 8ecfb1e..c3c8ad3 100644
--- a/installator.sh
+++ b/installator.sh
@@ -4,4 +4,11 @@ sudo apt-get install python-pip python-dev python-virtualenv
 targetDirectory=$1
 virtualenv --system-site-packages "./$targetDirectory"
 source ./$targetDirectory/bin/activate
-sudo -H pip install --upgrade tensorflow pip
+echo "Choose Python version:"
+echo "1 Python 2.7"
+echo "2 Python 3.n"
+read pythonVersion
+case $pythonVersion in
+  1) sudo -H pip install --upgrade tensorflow pip;;
+  2) sudo -H pip3 install --upgrade tensorflow pip;;
+esac