/bin/sh

TAO_TELEMETRY_SERVER=https://telemetry.metropolis.nvidia.com:443/api/v1/telemetry

/bin/bash

NVIDIA_TAO_TOOLKIT_VERSION=4.0.0-TF2

TAO_TOOLKIT_VERSION=4.0.0

NVIDIA_PRODUCT_NAME=TAO Toolkit

pip install --no-cache-dir git+https://github.com/cliffwoolley/jupyter_tensorboard.git@0.2.0+nv21.03 &&
  mkdir -p $NVM_DIR &&
  curl -Lo- https://raw.githubusercontent.com/nvm-sh/nvm/v0.38.0/install.sh | bash &&
  source "$NVM_DIR/nvm.sh" &&
  nvm install 16.18.1 --reinstall-packages-from=16.15.1 node &&
  nvm uninstall 16.15.1 &&
  jupyter labextension install jupyterlab_tensorboard &&
  jupyter serverextension enable jupyterlab &&
  pip install --no-cache-dir jupytext &&
  jupyter labextension install jupyterlab-jupytext@1.2.2 &&
  ( cd /usr/local/share/jupyter/lab/staging &&
  npm prune --production ) &&
  npm cache clean --force &&
  rm -rf /usr/local/share/.cache &&
  echo "source $NVM_DIR/nvm.sh" >> /etc/bash.bashrc &&
  mv /root/.jupyter/jupyter_notebook_config.json /usr/local/etc/jupyter/ &&
  jupyter lab clean

NVM_DIR=/usr/local/nvm

python -m pip uninstall graphviz -y

python -m pip install --upgrade joblib jupyter-core pillow mpmath