mirror of
https://github.com/gryf/coach.git
synced 2026-02-13 04:15:45 +01:00
coach v0.8.0
This commit is contained in:
Gal Leibovich
207
install.sh
Executable file
207
install.sh
Executable file
@@ -0,0 +1,207 @@
|
||||
#!/bin/bash -e
|
||||
|
||||
if [ "${HTTPS_PROXY}" == "" ]; then
|
||||
HTTPS_PROXY="${https_proxy}"
|
||||
fi
|
||||
|
||||
prompt () {
|
||||
# prints a yes / no question to the user and returns the answer
|
||||
# first argument is the prompt question
|
||||
# second argument is the default answer - Y / N
|
||||
local default_answer
|
||||
|
||||
# set the default value
|
||||
case "${2}" in
|
||||
y|Y ) default_answer=1; options="[Y/n]";;
|
||||
n|N ) default_answer=0; options="[y/N]";;
|
||||
"" ) default_answer=; options="[y/n]";;
|
||||
* ) echo "invalid default value"; exit;;
|
||||
esac
|
||||
|
||||
while true; do
|
||||
# read the user choice
|
||||
read -p "${1} ${options} " choice
|
||||
|
||||
# return the choice or the default value if an enter was pressed
|
||||
case "${choice}" in
|
||||
y|Y ) retval=1; return;;
|
||||
n|N ) retval=0; return;;
|
||||
"" ) if [ ! -z "${default_answer}" ]; then retval=${default_answer}; return; fi;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
add_to_bashrc () {
|
||||
# adds an env variable to the bashrc
|
||||
# first argument is the variable name
|
||||
# second argument is the variable value
|
||||
|
||||
EXISTS_IN_BASHRC=`awk '/${2}/{print $1}' ~/.bashrc`
|
||||
if [ "${EXISTS_IN_BASHRC}" == "" ]; then
|
||||
echo "export ${1}=${2}" >> ~/.bashrc
|
||||
fi
|
||||
}
|
||||
|
||||
GET_PREFERENCES_MANUALLY=1
|
||||
|
||||
INSTALL_COACH=0
|
||||
INSTALL_DASHBOARD=0
|
||||
INSTALL_GYM=0
|
||||
INSTALL_VIRTUAL_ENVIRONMENT=1
|
||||
|
||||
# Get user preferences
|
||||
TEMP=`getopt -o cpgvrmeNndh \
|
||||
--long coach,dashboard,gym,no_virtual_environment,ngraph,debug,help \
|
||||
-- "$@"`
|
||||
eval set -- "$TEMP"
|
||||
while true; do
|
||||
#for i in "$@"
|
||||
case ${1} in
|
||||
-c|--coach)
|
||||
INSTALL_COACH=1
|
||||
GET_PREFERENCES_MANUALLY=0
|
||||
shift;;
|
||||
-p|--dashboard)
|
||||
INSTALL_DASHBOARD=1
|
||||
GET_PREFERENCES_MANUALLY=0;
|
||||
shift;;
|
||||
-g|--gym)
|
||||
INSTALL_GYM=1
|
||||
GET_PREFERENCES_MANUALLY=0;
|
||||
shift;;
|
||||
-N|--no_virtual_environment)
|
||||
INSTALL_VIRTUAL_ENVIRONMENT=0
|
||||
GET_PREFERENCES_MANUALLY=0;
|
||||
shift;;
|
||||
-ng|--ngraph)
|
||||
INSTALL_NGRAPH=1
|
||||
GET_PREFERENCES_MANUALLY=0;
|
||||
shift;;
|
||||
-d|--debug) set -x; shift;;
|
||||
-h|--help)
|
||||
echo "Available command line arguments:"
|
||||
echo ""
|
||||
echo " -c | --coach - Install Coach requirements"
|
||||
echo " -p | --dashboard - Install Dashboard requirements"
|
||||
echo " -g | --gym - Install Gym support"
|
||||
echo " -N | --no_virtual_environment - Do not install inside of a virtual environment"
|
||||
echo " -d | --debug - Run in debug mode"
|
||||
echo " -h | --help - Display this help message"
|
||||
echo ""
|
||||
exit;;
|
||||
--) shift; break;;
|
||||
*) break;; # unknown option;;
|
||||
esac
|
||||
done
|
||||
|
||||
if [ ${GET_PREFERENCES_MANUALLY} -eq 1 ]; then
|
||||
prompt "Install Coach requirements?" Y
|
||||
INSTALL_COACH=${retval}
|
||||
|
||||
prompt "Install Dashboard requirements?" Y
|
||||
INSTALL_DASHBOARD=${retval}
|
||||
|
||||
prompt "Install Gym support?" Y
|
||||
INSTALL_GYM=${retval}
|
||||
|
||||
prompt "Install Neon and NGraph support?" Y
|
||||
INSTALL_NGRAPH=${retval}
|
||||
fi
|
||||
|
||||
IN_VIRTUAL_ENV=`python -c 'import sys; print("%i" % hasattr(sys, "real_prefix"))'`
|
||||
|
||||
# basic installations
|
||||
sudo -E apt-get install python3-pip cmake zlib1g-dev python3-tk python-opencv -y
|
||||
pip3 install --upgrade pip
|
||||
|
||||
# if we are not in a virtual environment, we will create one with the appropriate python version and then activate it
|
||||
# if we are already in a virtual environment,
|
||||
|
||||
if [ ${INSTALL_VIRTUAL_ENVIRONMENT} -eq 1 ]; then
|
||||
if [ ${IN_VIRTUAL_ENV} -eq 0 ]; then
|
||||
sudo -E pip3 install virtualenv
|
||||
virtualenv -p python3 coach_env
|
||||
. coach_env/bin/activate
|
||||
fi
|
||||
fi
|
||||
|
||||
#------------------------------------------------
|
||||
# From now on we are in a virtual environment
|
||||
#------------------------------------------------
|
||||
|
||||
# get python local and global paths
|
||||
python_version=python$(python -c "import sys; print (str(sys.version_info[0])+'.'+str(sys.version_info[1]))")
|
||||
var=( $(which -a $python_version) )
|
||||
get_python_lib_cmd="from distutils.sysconfig import get_python_lib; print (get_python_lib())"
|
||||
lib_virtualenv_path=$(python -c "$get_python_lib_cmd")
|
||||
lib_system_path=$(${var[-1]} -c "$get_python_lib_cmd")
|
||||
|
||||
# Boost libraries
|
||||
sudo -E apt-get install libboost-all-dev -y
|
||||
|
||||
# Coach
|
||||
if [ ${INSTALL_COACH} -eq 1 ]; then
|
||||
echo "Installing Coach requirements"
|
||||
pip install -r ./requirements_coach.txt --proxy ${HTTPS_PROXY}
|
||||
fi
|
||||
|
||||
# Dashboard
|
||||
if [ ${INSTALL_DASHBOARD} -eq 1 ]; then
|
||||
echo "Installing Dashboard requirements"
|
||||
pip install -r ./requirements_dashboard.txt --proxy ${HTTPS_PROXY}
|
||||
sudo -E apt-get install dpkg-dev build-essential python3.5-dev libjpeg-dev libtiff-dev libsdl1.2-dev libnotify-dev \
|
||||
freeglut3 freeglut3-dev libsm-dev libgtk2.0-dev libgtk-3-dev libwebkitgtk-dev libgtk-3-dev libwebkitgtk-3.0-dev libgstreamer-plugins-base1.0-dev -y
|
||||
|
||||
sudo -E -H pip3 install -U --pre -f \
|
||||
https://wxpython.org/Phoenix/snapshot-builds/linux/gtk3/ubuntu-16.04/wxPython-4.0.0a3.dev3059+4a5c5d9-cp35-cp35m-linux_x86_64.whl wxPython
|
||||
|
||||
# link wxPython Phoenix library into the virtualenv since it is installed with apt-get and not accessible
|
||||
libs=( wx )
|
||||
for lib in ${libs[@]}
|
||||
do
|
||||
ln -s $lib_system_path/$lib $lib_virtualenv_path/$lib
|
||||
done
|
||||
fi
|
||||
|
||||
# Gym
|
||||
if [ ${INSTALL_GYM} -eq 1 ]; then
|
||||
echo "Installing Gym support"
|
||||
sudo -E apt-get install xvfb libav-tools xorg-dev libsdl2-dev swig cmake -y
|
||||
pip install box2d # for bipedal walker etc.
|
||||
pip install gym
|
||||
fi
|
||||
|
||||
# NGraph and Neon
|
||||
if [ ${INSTALL_NGRAPH} -eq 1 ]; then
|
||||
echo "Installing NGraph requirements"
|
||||
|
||||
# MKL
|
||||
git clone https://github.com/01org/mkl-dnn.git
|
||||
cd mkl-dnn
|
||||
cd scripts && ./prepare_mkl.sh && cd ..
|
||||
mkdir -p build && cd build && cmake .. && make -j
|
||||
sudo make install -j
|
||||
cd ../..
|
||||
export MKLDNN_ROOT=/usr/local/
|
||||
add_to_bashrc MKLDNN_ROOT ${MKLDNN_ROOT}
|
||||
export LD_LIBRARY_PATH=$MKLDNN_ROOT/lib:$LD_LIBRARY_PATH
|
||||
add_to_bashrc LD_LIBRARY_PATH ${MKLDNN_ROOT}/lib:$LD_LIBRARY_PATH
|
||||
|
||||
# NGraph
|
||||
git clone https://github.com/NervanaSystems/ngraph.git
|
||||
cd ngraph
|
||||
make install -j
|
||||
make gpu_prepare -j
|
||||
cd ..
|
||||
|
||||
# Neon
|
||||
sudo -E apt-get install libhdf5-dev libyaml-dev pkg-config clang
|
||||
git clone https://github.com/NervanaSystems/neon.git
|
||||
cd neon && make sysinstall -j
|
||||
cd ..
|
||||
fi
|
||||
|
||||
# Intel Optimized TensorFlow
|
||||
pip3 install https://anaconda.org/intel/tensorflow/1.3.0/download/tensorflow-1.3.0-cp35-cp35m-linux_x86_64.whl
|
||||
|
||||
|
||||
Reference in New Issue
Block a user