Added Python 3.n support

ml-in-programming · Mar 7, 2017 · 3903a5d · 3903a5d
1 parent 194e0a3
commit 3903a5d
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/installator.sh b/installator.sh
@@ -4,4 +4,15 @@ 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