Add the Nsight Jupyter plugin.

Author: brycelelbach

build/docker-compose.yml

@@ -20,7 +20,11 @@ services:
       - "0.0.0.0:8888:8888" # JupyterLab
     volumes:
       - /home/ubuntu/pyhpc-tutorial:/pyhpc-tutorial
+      - /var/run/docker.sock:/var/run/docker.sock
+      - /usr/bin/docker:/usr/bin/docker
     user: root
+    group_add:
+      - docker
     working_dir: /pyhpc-tutorial/notebooks
     command: ["/bin/bash", "-lc", "exec python -m jupyter lab --allow-root --ip=0.0.0.0 --no-browser --NotebookApp.token='' --NotebookApp.password='' --NotebookApp.default_url=''"]
     restart: unless-stopped

build/dockerfile

@@ -7,6 +7,8 @@ WORKDIR /pyhpc-tutorial/notebooks
 RUN pip install --root-user-action ignore -r /pyhpc-tutorial/build/requirements.txt \
  && git config --unset-all "http.https://github.com/.extraheader" || { code=$?; [ "$code" = 5 ] || exit "$code"; } \
  && git config --global --add safe.directory "/pyhpc-tutorial" \
- && mkdir -p ~/.jupyter \
+ && mkdir ~/.jupyter \
  && mkdir -p ~/.local/state/._bash_history \
- && ln -fs /pyhpc-tutorial/build/jupyter_server_config.py ~/.jupyter/jupyter_server_config.py
+ && ln -fs /pyhpc-tutorial/build/jupyter_server_config.py ~/.jupyter/jupyter_server_config.py \
+ && ln -fs /pyhpc-tutorial/build/jupyter_nsight_plugin_settings.json ~/.jupyter/lab/user-settings/jupyterlab-nvidia-nsight/plugin.jupyterlab-settings
+

build/jupyter_nsight_plugin_settings.json

@@ -0,0 +1,13 @@
+{
+  "ui": {
+    "enabled": true,
+    "suppressServerAddressWarning": true,
+  },
+  "nsys": {
+    "installationPath": "/usr/local/cuda",
+    "args": "--trace=cuda,nvtx,osrt --python-sampling=true --python-backtrace=cuda --cudabacktrace=all",
+  },
+  "ncu": {
+    "installationPath": "/usr/local/cuda",
+  }
+}

build/requirements.txt

@@ -7,6 +7,7 @@ opencv-python
 
 # Jupyter
 jupyter
+jupyterlab-nvidia-nsight
 
 # CUDA
 cuda-python == 12.9.1 # cuda.{core, bindings, cooperative, parallel}, numba.cuda