Notebook Runner

This example shows how to run an entire Jupyter Notebook on Robbie as a long-running training job, instead of an individual function.

The Notebook Runner is great for when you are done experimenting in an interactive notebook and want to execute a long-running training job that could take hours or days.

To run your first notebook job on Robbie, open a command prompt and type:

jupyter lab

Create two new cells in the notebook.

In the first cell paste:

import robbie
robbie.init()

In the second cell paste:

import os
os.system('nvidia-smi')
  • The first cell imports the Robbie Python package and then initializes it. If you have not previously logged in to Robbie, you will be prompted for your username and password.

  • The second cell contains the Python code that will be run in the Robbie cloud.

Save the notebook ( on a Mac or on Windows)

Run the first cell by pressing the button.

You should see a small UI directly in the notebook.

Select your desired Hardware, from the dropdown and tick the Tail the remote machine stdout check box.

Next, press the button.

You should then see the command program executing.

A screenshot of a computer program

So, what did Robbie do?

  1. Converted your notebook to a Python .py file and removed any notebook-specific constructs like ! %, or %%.

  2. Automatically determined the Python dependencies from the .py file and generated a requirement.txt file.

  3. Selected a default GPU compute instance (NERC Small) in the Robbie cloud

  4. Set billing for the job to your `Personal` funds (including free trial)

  5. Create a new ‘run’ called intermediate_macaw

  6. Copied the local workspace file, including the .py to the remote machine

  7. Vended a deep link so you can track the progress of the run in the Robbie portal

  8. Launched the run in the Robbie cloud

  9. Printed out the duration and cost of the job once it was complete

For more examples, see Jupyter Notebook Examples.

Last updated

Was this helpful?