#!/bin/bash

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

#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 update && \
apt-get install -y ipmitool && modprobe ipmi_devintf && \
apt-get install -y 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 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 

#check nvidia drivers
nvidia-smi

echo "Add sudo users with: usermod -aG sudo USER"