Set Up Your Notebooks

Getting started with Jupyter notebooks on Kubeflow

Your Kubeflow deployment includes services for spawning and managing Jupyternotebooks.

You can set up multiple notebook servers per Kubeflow deployment. Eachnotebook server can include multiple notebooks. Each notebook server belongsto a single namespace, which corresponds to the project group or team for thatserver.

This guide shows you how to set up a notebook server for your Jupyter notebooksin Kubeflow.

Quick guide

Summary of steps:

  • Follow the Kubeflow getting-started guideto set up your Kubeflow deployment and open the Kubeflow UI.

  • Click Notebook Servers in the left-hand panel of the Kubeflow UI.

  • Choose the namespace corresponding to your Kubeflow profile.

  • Click NEW SERVER to create a notebook server.

  • When the notebook server provisioning is complete, click CONNECT.

  • Click Upload to upload an existing notebook, or click New tocreate an empty notebook.

The rest of this page contains details of the above steps.

Install Kubeflow and open the Kubeflow UI

Follow the Kubeflow getting-started guide toset up your Kubeflow deployment in your environment of choice (locally, onpremises, or in the cloud).

When Kubeflow is running, access the Kubeflow UI as described in thegetting-started guide for your chosen environment. For example:

  • If you deployed Kubeflow on Google Cloud Platform (GCP), the Kubeflow UI isavailable at the following URI:
  1. https://<deployment_name>.endpoints.<project>.cloud.goog/
  • If you set up port forwarding to the Ambassador service, the Kubeflow UI isavailable at the following URI:
  1. http://localhost:8080/
  • For other environments, see the getting-started guide for your chosenenvironment.

Create a Jupyter notebook server and add a notebook

  • Click Notebook Servers in the left-hand panel of the Kubeflow UI toaccess the Jupyter notebook services deployed with Kubeflow:Opening notebooks from the Kubeflow UI

  • Sign in:

    • On GCP, sign in using your Google Account. (If you have already logged into your Google Account you may not need to log in again.)
    • On all other platforms, sign in using any username and password.

  • Select a namespace:

    • Click the namespace dropdown to see the list of available namespaces.
    • Choose the namespace that corresponds to your Kubeflow profile. (Seethe page on multi-user isolationfor more information about namespaces.) Selecting a Kubeflow namespace

  • Click NEW SERVER on the Notebook Servers page:

The Kubeflow notebook servers page

You should see a page for entering details of your new server. Here is apartial screenshot of the page:

Form for adding a Kubeflow notebook server

  • Enter a name of your choice for the notebook server. The name caninclude letters and numbers, but no spaces. For example, my-first-notebook.

  • Kubeflow automatically updates the value in the namespace field tobe the same as the namespace that you selected in a previous step. Thisensures that the new notebook server is in a namespace that you can access.

  • Select a Docker image for the baseline deployment of your notebookserver. You can choose from a range of standard images or specify acustom image:

    • Standard: The standard Docker images include typical machine learning(ML) packages that you can use within your Jupyter notebooks onthis notebook server. Select an image from the Image dropdown menu.The image names indicate the following choices:

      • A TensorFlow version (for example, tensorflow-1.13.1). Kubeflow offersa CPU and a GPU image for each minor version of TensorFlow.

      • cpu or gpu, depending on whether you want to train your model on a CPUor a GPU.

        • If you choose a GPU image, make sure that you have GPUsavailable in your Kubeflow cluster. Run the following command to checkif there are any GPUs available:kubectl get nodes "-o=custom-columns=NAME:.metadata.name,GPU:.status.allocatable.nvidia.com/gpu"
        • If you have GPUs available, you can schedule your server on a GPU nodein the Extra Resources section at the bottom of the form. Forexample, to reserve two GPUs, enter the following JSON code:{"nvidia.com/gpu": 2}
      • Kubeflow version (for example, v0.5.0).

    • Custom: If you select the custom option, you must specify a Docker imagein the form registry/image:tag. For guidelines on creating a Dockerimage for your notebook, see the guide tocreating a custom Jupyter image.

  • Specify the total amount of CPU that your notebook server should reserve.The default is 0.5. For CPU-intensive jobs, you can choose more than one CPU(for example, 1.5).

  • Specify the total amount of memory (RAM) that your notebook server shouldreserve. The default is 1.0Gi.

  • Specify a workspace volume to hold your personal workspace for thisnotebook server. Kubeflow provisions aKubernetes persistent volume (PV) for your workspace volume. The PV ensures that you canretain data even if you destroy your notebook server.

    • The default is to create a new volume for your workspace with thefollowing configuration:

      • Name: The volume name is synced with the name of the notebook server.When you start typing the notebook server name, the volume name takesthe same value. You can edit the volume name, but if you later edit thenotebook server name, the volume name changes to match the notebookserver name.
      • Size: 10Gi
      • Mount path: /home/jovyan
      • Access mode: ReadWriteOnce. This setting means that the volume can bemounted as read-write by a single node. See theKubernetes documentation for more details about access modes.
    • Alternatively, you can point the notebook server at an existing volume byspecifying the name, mount path, and access mode for the existing volume.

  • (Optional) Specify one or more data volumes if you want to store andaccess data from the notebooks on this notebook server. You can add newvolumes or specify existing volumes. Kubeflow provisions aKubernetes persistent volume (PV) for each of your data volumes.

  • Click LAUNCH. You should see an entry for your newnotebook server on the Notebook Servers page, with a spinning indicator inthe Status column. It can take a few minutes to set upthe notebook server.

    • You can check the status of your Pod by hovering your mouse cursor overthe icon in the Status column next to the entry for your notebookserver. For example, if the image is downloading then the status spinnerhas a tooltip that says ContainerCreating.

Alternatively, you can check the Pod status by entering the followingcommand:

  1. kubectl -n <NAMESPACE> describe pods jupyter-<USERNAME>

Where <NAMESPACE> is the namespace you specified earlier(default kubeflow) and <USERNAME> is the name you used to log in.A note for GCP users: If you have IAP turned on, the Pod hasa different name. For example, if you signed in as USER@DOMAIN.EXTthe Pod has a name of the following form:

  1. jupyter-accounts-2egoogle-2ecom-3USER-40DOMAIN-2eEXT
  • When the notebook server provisioning is complete, you should see an entryfor your server on the Notebook Servers page, with a check mark in theStatus column:

Opening notebooks from the Kubeflow UI

  • Click CONNECT to start the notebook server.

  • When the notebook server is running, you should see the Jupyter dashboardinterface. If you requested a new workspace, the dashboard should be emptyof notebooks:

Jupyter dashboard with no notebooks

  • Click Upload to upload an existing notebook, or click New tocreate an empty notebook. You can read about using notebooks in theJupyter documentation.

Experiment with your notebook

The default notebook image includes all the plugins that you need to train aTensorFlow model with Jupyter, includingTensorboardfor rich visualizations and insights into your model.

To test your Jupyter installation, you can run a basic ‘hello world’ program(adapted frommnist_softmax.py) as follows:

  • Use the Jupyter dashboard to create a new Python 3 notebook.

  • Copy the following code and paste it into a code block in your notebook:

  1. from tensorflow.examples.tutorials.mnist import input_data
  2. mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
  4. import tensorflow as tf
  6. x = tf.placeholder(tf.float32, [None, 784])
  8. W = tf.Variable(tf.zeros([784, 10]))
  9. b = tf.Variable(tf.zeros([10]))
  11. y = tf.nn.softmax(tf.matmul(x, W) + b)
  13. y_ = tf.placeholder(tf.float32, [None, 10])
  14. cross_entropy = tf.reduce_mean(-tf.reduce_sum(y_ * tf.log(y), reduction_indices=[1]))
  16. train_step = tf.train.GradientDescentOptimizer(0.05).minimize(cross_entropy)
  18. sess = tf.InteractiveSession()
  19. tf.global_variables_initializer().run()
  21. for _ in range(1000):
  22.   batch_xs, batch_ys = mnist.train.next_batch(100)
  23.   sess.run(train_step, feed_dict={x: batch_xs, y_: batch_ys})
  25. correct_prediction = tf.equal(tf.argmax(y,1), tf.argmax(y_,1))
  26. accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
  27. print("Accuracy: ", sess.run(accuracy, feed_dict={x: mnist.test.images, y_: mnist.test.labels}))
  • Run the code. You should see a number of WARNING messages from TensorFlow,followed by a line showing a training accuracy something like this:
  1. Accuracy:  0.9012

Please note that when running on most cloud providers, the public IP address isexposed to the internet and is an unsecured endpoint by default.

Next steps