Skip to content

Jupyter Overview

Overview

The Jupyter Notebook is a web application that enables you to create and share documents (called "notebooks") that can contain a mix of live code, equations, visualizations, and explanatory text.

This is an introduction to using these notebooks on Savio. Before getting started, make sure you have access to the Savio cluster, as you will need your BRC username and one-time password to log in.

As described next, you can start a Jupyter notebook via the Open OnDemand service, which allows you to operate completely via your web browser on your local computer (e.g., your laptop). Open OnDemand replaces our old (much more limited) JupyterHub service.

Using Jupyter notebooks via the interactive Open OnDemand service

Running a notebook via Open OnDemand

  1. Connect to https://ood.brc.berkeley.edu
  1. Just after logging in with your BRC username and one-time password (OTP), the initial OnDemand screen presents a welcome screen. Click the "Interactive Apps" pulldown.

    Screenshot: OOD Interactive Apps

  1. Choose the appropriate Jupyter Server option to either run as part of a Slurm batch job ("compute via Slurm in Slurm partitions") or (for non-intensive computations) on our standalone Open OnDemand server ("compute on shared Jupyter node").

    Screenshot: OOD Jupyter Interactive Apps Choices

  1. Fill out the form presented to you and click on "Launch". For Slurm-based work, you'll need to provide the information that you would usually supply when submitting a Slurm job. Once the server is ready, you'll be able to click no the "Connect to Jupyter" button.

    Screenshot: OOD Jupyter Form

    Screenshot: OOD Jupyter Submit

    Screenshot: OOD Jupyter Connect

  1. After clicking on "Connect to Jupyter", the home directory will be displayed. From the "New" dropdown menu (next to 'Upload' near the top right of the screen) select one of the options listed, which include:
    1. Select "Python 3.10.10 (TF-2.12.0)" under "Notebooks" for a Notebook with full access to Python 3.10.10 and the system-installed Python 3.10.10 packages (Python 3.10.10 and its packages are also available through the python/3.10.10 module when logging into Savio in the usual way), including Tensorflow 2.12.0.
    2. Select "Python 3.9.12" under "Notebooks" for a Notebook with full access to Python 3.9.12 and the system-installed Python 3.9.12 packages (Python 3.9.12 and its packages are also available through the python/3.9.12 module when logging into Savio in the usual way).
    3. Select “Terminal” to open a UNIX terminal session instead, so that you can work at the command line, rather than in a Notebook. (You might select this option if you need to set up IPython Clusters or add kernels, as described further below.)

    Screenshot: OOD Jupyter New

  1. To move between your Notebook and the control page that allows you to see your files and select any running Notebooks, clusters and terminals, simply click on the ‘jupyter’ banner in the upper left corner of your Notebook. (Or alternately, select “Control Panel” and then click “My Server”.)

    Screenshot: OOD Jupyter Banner

  1. You can have your session continue to operate in the background by selecting the “Logout” button. Note that, when doing so, you will continue to be charged for FCA use or, if using a Condo node, you will prevent others in your group from accessing the relevant resource on which the job is running.

    Screenshot: OOD Jupyter Logout

  1. To terminate a running Notebook, select the “My Interactive Sessions” tab and delete the session.

    Screenshot: OOD Jupyter Logout

At this point you should already have a fully working Jupyter environment. To start working with Jupyter Notebooks, please see the Jupyter Documentation.

To use parallelization capabilities via your notebook, please see Running Parallel Python Code in Jupyter Notebooks on Savio.

Installing Python packages

A variety of standard Python packages (such as numpy, scipy, matplotlib and pandas) are available automatically. To see what packages are available, open a Terminal notebook (see item 5c above) or open a Terminal on Savio in the usual fashion. Then load the Python version of interest (here Python 3.10.10) and list the installed packages:

module load python/3.10.10
conda list

There should be no issues using pip to install or upgrade packages and then use them in a Jupyter notebook, but you will need to make sure to install the new versions or additional packages in your home or scratch directories because you do not have write permissions to the module directories. You can use pip install --user $MODULENAME to install the module to $HOME/.local. So, if you need to install additional packages, simply load the desired Python module in the usual way and then use pip to install in your home directory. For example, you can install the rpy2 package (needed in the next section) with:

module load python/3.10.10
pip install --user rpy2

If you'd like to install packages with conda install you'll need to create a Conda environment in which to install packages and then create a kernel associated with your Conda environment as discussed in the next section.

Adding new kernels

Jupyter supports notebooks in dozens of languages, including IPython, R, Julia, etc.

If you’d like to use a language not indicated in the drop-down menu discussed in step 6 above, you’ll need to create your own kernel. You may also need to create your own kernel for a language already supported if you want to customize your environment. For example, to set UNIX environment variables (such as $PYTHONPATH if you have packages installed in non-standard locations) or source a script in advance of running your notebook, you can do this by creating your own kernel. Or if you'd like to work within a Conda environment when using your notebook, you'll also need to create a kernel.

Using a Conda environment

Another approach to adding a new (Python) kernel  to your Jupyter environment is to create a conda environment and add it as a kernel to Jupyter. When in Jupyter, you will then be able to select the name from the kernel list, and it will be using the packages you installed. Follow these steps to do this (replacing $ENV_NAME with the name you want to give your conda environment): 

module load python/3.10.10
conda create --name=$ENV_NAME ipykernel
source activate $ENV_NAME
python -m ipykernel install --user --name $ENV_NAME
# Then from here you can do pip install or conda install

As a specific example, here we’ll illustrate how to create your own Python kernel within a Python Jupyter environment, using a specific Python version and specific numpy version. In this case I use `mamba`, a drop-in replacement for `conda` that is usually faster at installing packages.

module load python/3.10.10
conda create --name=my_numpy -c conda-forge python=3.11 ipykernel
source activate my_numpy
python -m ipykernel install --user --name my_numpy --display-name="Numpy 1.25.3"
mamba install -c conda-forge numpy=1.25.1

Now you can choose the kernel you just created from the kernel list in your Jupyter environment, and you can verify that you can utilize and access the specified numpy package from within a cell in a Jupyter notebook as follows:

import numpy
numpy.__version__

Conda environments can take up a lot of space

If you don't need a different version of Python from what the system provides, then using a virtualenv may take less space on disk than a conda environment. Just create the virtualenv, activate it, run pip install ipykernel, and then run the python -m ipykernel install command as above.

Manually creating a new kernel

In some cases you may need to manually create or edit a kernel. One situation is when your notebook needs to make use of software that is available via one of Savio's software modules.

Say we want to create an kernel that uses the ml/tensorflow/2.5.0-py37 module.

module load ml/tensorflow/2.5.0-py37
python -m ipykernel install --user --name=tf25 --display-name="Python3.7 TF2.5.0"

This creates ~/.local/jupyter/kernels/tf25/kernel.json:

{
 "argv": [
  "/global/software/sl-7.x86_64/modules/langs/python/3.7/bin/python",
  "-m",
  "ipykernel_launcher",
  "-f",
  "{connection_file}"
 ],
 "display_name": "Python3.7 TF-2.5.0",
 "language": "python"
}
We'll need to edit this file so that the kernel has access to module and its dependencies. We'll add a dictionary called "env" with environment variables to set when starting up the kernel. To know what to add in this case, take a look at your environment variables after you run module load ml/tensorflow/2.5.0-py37:

echo $PATH
echo $LD_LIBRARY_PATH
echo $PYTHONPATH

The final kernel.json will look like this:

{
    "argv": [
        "/global/software/sl-7.x86_64/modules/langs/python/3.7/bin/python",
        "-m",
        "ipykernel_launcher",
        "-f",
        "{connection_file}"
    ],
    "display_name": "Python3.7 TF-2.5.0",
    "language": "python",
    "env": {
        "PATH": "/global/software/sl-7.x86_64/modules/apps/ml/tensorflow/2.5.0-py37/bin:/global/software/sl-7.x86_64/modules/langs/cuda/11.2/bin:/global/software/sl-7.x86_64/modules/langs/python/3.7/bin:/global/software/sl-7.x86_64/modules/langs/python/3.7/condabin:/usr/local/bin:/usr/bin:/usr/local/sbin:/usr/sbin:/global/home/groups/allhands/bin:/global/home/users/jpduncan/bin",
        "LD_LIBRARY_PATH": "/global/software/sl-7.x86_64/modules/cuda/11.2/cudnn/8.1.1/lib64:/global/software/sl-7.x86_64/modules/langs/cuda/11.2/lib64/stubs:/global/software/sl-7.x86_64/modules/langs/cuda/11.2/lib64",
        "PYTHONPATH" : "/global/software/sl-7.x86_64/modules/apps/ml/tensorflow/2.5.0-py37/lib/python3.7/site-packages"
    }
}
Now I can use my new TensorFlow kernel on a GPU node.

Please review the IPython Kernel Specs for more details regarding the format and contents of this configuration file. In particular, please make sure $PATH$LD_LIBRARY_PATH$PYTHONPATH, and all other environment variables that you use in the kernel are properly populated with the correct values.

Enabling extensions

If you want to create or use notebooks with interactive widgets, a table of contents, or collapsible code blocks, you need to enable Nbextensions. From the “New” dropdown menu (same as in step 4), select “Terminal”. Copy and paste the code below into the terminal prompt, and hit enter.

module load python/3.7
pip install jupyter_contrib_nbextensions --user
jupyter contrib nbextension install --user

After stopping and restarting your server as well as logging out and back in, when you return to the JupyterHub page, you should see a new tab for Nbextensions, where you can enable or disable individual extensions. Shut down and relaunch any running notebooks, and the extensions will be present.

Using the Jupyter Lab interface

Once you've started the Jupyter interface and see the main Jupyter page, which should show a list of files in your working directory, modify the URL in your browser so it ends in "/lab" rather than "/tree?". That will change to the Jupyter Lab interface.

Note that to exit your session you'll need to switch back to "/tree" and stop your server as described in the relevant previous section.