Updated Getting Started section

Author: prabhuomkar

README.md

@@ -4,18 +4,14 @@
 
 This repository provides tutorial code in C++ for deep learning researchers to learn PyTorch.
 
-## How to Build
+## Getting Started
 - Install
 ```
-wget https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip
-unzip libtorch-shared-with-deps-latest.zip
+./scripts.sh install
 ```
 - Build
 ```
-mkdir build
-cd build
-cmake -DCMAKE_PREFIX_PATH=/absolute/path/to/libtorch ..
-make
+./scripts.sh build
 ```
 
 ## Table of Contents

scripts.sh

@@ -0,0 +1,22 @@
+#!/bin/bash
+
+function install() {
+	wget https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip
+	unzip libtorch-shared-with-deps-latest.zip
+  rm -rf libtorch-shared-with-deps-latest.zip
+}
+
+function build() {
+	rm -rf build
+	mkdir build
+	cd build
+	cmake -DCMAKE_PREFIX_PATH=$(dirname $(pwd))/libtorch ..
+	make
+}
+
+if [ $1 = "install" ]
+then
+	install
+else
+	build
+fi