#!/bin/bash

if [ "$EUID" -ne 0 ]
  then echo "Please run as root"
  exit
fi

apt-get update

apt-get install -y --no-install-recommends libnvinfer7=7.1.3-1+cuda11.0 \
    libnvinfer-dev=7.1.3-1+cuda11.0 \
    libnvinfer-plugin7=7.1.3-1+cuda11.0


apt-get install sudo
apt-get install ipmitool && modprobe ipmi_devintf
apt-get install python3-pip nodejs npm git

python3 -m pip install jupyterhub
npm install -g configurable-http-proxy
python3 -m pip install jupyterlab notebook tensorflow jupyterlab-git
python3 -m pip install tqdm pickle pandas ipywidgets pillow matplotlib
python3 -m pip install scipy scikit-learn

mkdir /etc/jupyterhub/
cp jupyterhub_config.py /etc/jupyterhub/jupyterhub_config.py
cp jupyterhub.service /etc/systemd/system/jupyterhub.service

systemctl enable jupyterhub
systemctl start jupyterhub

#Add sudo users with: usermod -aG sudo USER