Update files

Author: saravanabalagi

src/jupyterlab.def

@@ -26,8 +26,8 @@ From: tf2.sif
   npm --version
 
   # jupyterlab and other pip dependencies
-  pip install "jupyterlab>=1,<2"
-  pip install "nbdime<2" jupyterlab-git
+  pip install jupyterlab
+  pip install nbdime jupyterlab-git
   pip install ipympl
 
   # jupyterlab extensions
@@ -36,11 +36,11 @@ From: tf2.sif
   jupyter labextension install @lckr/jupyterlab_variableinspector
   jupyter labextension install @jupyter-widgets/jupyterlab-manager jupyter-matplotlib
   jupyter labextension install @jupyterlab/toc
-  # jupyter labextension install @krassowski/jupyterlab_go_to_definition  #v2.0
+  jupyter labextension install @krassowski/jupyterlab_go_to_definition
+  jupyter labextension install jupyterlab-drawio
 
   # Older extensions (Supports only jupyterlab v1)
-  jupyter labextension install @krassowski/jupyterlab_go_to_definition@0.7.1
-  jupyter labextension install jupyterlab-drawio
+  # jupyter labextension install @krassowski/jupyterlab_go_to_definition@0.7.1
 
   # jupyterlab extensions config
   # Use with Git SSH; Does not work with HTTPS yet

src/tf2.def

@@ -1,5 +1,5 @@
 Bootstrap: docker
-From: tensorflow/tensorflow:2.1.0-gpu-py3
+From: tensorflow/tensorflow:2.2.0-gpu
 
 %post
 
@@ -19,8 +19,8 @@ From: tensorflow/tensorflow:2.1.0-gpu-py3
   pip install tensorflow-hub
 
   # with only pip install opencv will fail on import unable to link libs
+  apt update && apt install -y ffmpeg libsm6 libxext6 libxrender-dev
   pip install opencv-python
-  apt update && apt install -y libsm6 libxext6 libxrender-dev
 
   # Computer Vision
   pip install sklearn scikit-image
@@ -31,6 +31,7 @@ From: tensorflow/tensorflow:2.1.0-gpu-py3
   apt update && apt install -y graphviz
   pip install pydot
   pip install seaborn
+  pip install -U tensorboard_plugin_profile
 
   # Data Processing
   pip install pandas
@@ -50,6 +51,13 @@ From: tensorflow/tensorflow:2.1.0-gpu-py3
   apt update && apt install -y tmux
   apt install -y sysstat
 
+  # Install pid_usage
+  apt install wget
+  wget https://github.com/saravanabalagi/pid_usage/releases/download/0.1/pid_usage.tar.gz
+  tar -xzf pid_usage.tar.gz
+  mv pid_usage /bin
+  rm pid_usage.tar.gz
+
 %labels
 
   Author Saravanabalagi Ramachandran