Merge pull request #232 from mrdaliri/requirements-installer

Requirements installer

Author: FloopCZ

Dockerfiles/install-ubuntu.sh

@@ -13,25 +13,7 @@ for key in "$@"; do
 done
 
 # install requirements
-export DEBIAN_FRONTEND=noninteractive
-apt-get -y update
-apt-get -y install \
-  cmake \
-  g++-7 \
-  git \
-  python3-dev \
-  python3-numpy \
-  sudo \
-  wget
-
-# install bazel
-export BAZEL_VERSION=${BAZEL_VERSION:-`cat ./tensorflow_cc/Dockerfiles/BAZEL_VERSION`}
-apt-get -y install pkg-config zip g++ zlib1g-dev unzip python
-bazel_installer=bazel-${BAZEL_VERSION}-installer-linux-x86_64.sh
-wget -P /tmp https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/${bazel_installer}
-chmod +x /tmp/${bazel_installer}
-/tmp/${bazel_installer}
-rm /tmp/${bazel_installer}
+./tensorflow_cc/ubuntu-requirements.sh
 
 if $cuda; then
     # install libcupti

ubuntu-requirements.sh

@@ -0,0 +1,36 @@
+#!/bin/bash
+set -e
+
+if [[ $UID != 0 ]]; then
+    echo "Please run this script with sudo:"
+    echo "sudo $0 $*"
+    exit 1
+fi
+
+if [[ "$OSTYPE" == "linux-gnu" ]]; then
+    # install requirements
+    export DEBIAN_FRONTEND=noninteractive
+    apt-get -y update
+    apt-get -y install \
+    cmake \
+    g++-7 \
+    git \
+    python3-dev \
+    python3-numpy \
+    sudo \
+    wget
+
+    # install bazel
+    export BAZEL_VERSION=${BAZEL_VERSION:-`cat $(dirname "$0")/Dockerfiles/BAZEL_VERSION`}
+    apt-get -y install pkg-config zip g++ zlib1g-dev unzip python
+    bazel_installer=bazel-${BAZEL_VERSION}-installer-linux-x86_64.sh
+    wget -P /tmp https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/${bazel_installer}
+    chmod +x /tmp/${bazel_installer}
+    /tmp/${bazel_installer}
+    rm /tmp/${bazel_installer}
+
+else
+    echo "This script supports only Debian-based operating systems (like Ubuntu)." \
+         "Please consult README file for manual installation on your '$OSTYPE' OS."
+    exit 1
+fi