Run on Google Colab

With Google Colaboratory, we can easily set up and run code in the cloud. This page illustrates the steps to install Analytics Zoo and run notebooks on colaboratory.

First, create or load a notebook file in colaboratory. Then, prepare the environment. You only need to install JDK and Analytics Zoo. As installing analytics-zoo from pip will automatically install pyspark, you are recommended not to install pyspark by yourself.

Prepare Environment

Install Java 8

Run the command on the colaboratory file to install jdk 1.8:

# Install jdk8
!apt-get install openjdk-8-jdk-headless -qq > /dev/null
# Set jdk environment path which enables you to run Pyspark in your Colab environment.
import os
os.environ["JAVA_HOME"] = "/usr/lib/jvm/java-8-openjdk-amd64"
!update-alternatives --set java /usr/lib/jvm/java-8-openjdk-amd64/jre/bin/java

Install Analytics Zoo from pip

You can add the following command on your colab file to install the analytics-zoo via pip easily:

# Install latest release version of analytics-zoo 
# Installing analytics-zoo from pip will automatically install pyspark, bigdl, and their dependencies.
!pip install analytics-zoo

Begin your code

Call init_nncontext() that will create a SparkContext with optimized performance configurations.

from zoo.common.nncontext import*

sc = init_nncontext()

Output on Colaboratory:

Prepending /usr/local/lib/python3.6/dist-packages/bigdl/share/conf/spark-bigdl.conf to sys.path
Adding /usr/local/lib/python3.6/dist-packages/zoo/share/lib/analytics-zoo-bigdl_0.10.0-spark_2.4.3-0.7.0-jar-with-dependencies.jar to BIGDL_JARS
Prepending /usr/local/lib/python3.6/dist-packages/zoo/share/conf/spark-analytics-zoo.conf to sys.path

Run Github Notebook on colaboratory

If you would like to open Analytics Zoo Notebook in a GitHub repo directly, the only thing you need to do is:

For example, change the URL of Analytics Zoo tutorial to

Then, prepare the environment of Java8 and Analytics Zoo as described above at the beginning of the colab notebook. If you would like to save the changes, you can make a copy to drive and run it within the instructions.


As Colab engine provides your application with some built-in Python libraries, you should check if the library versions are compatible with your application. You may refer this document to specify the python library version that Analytics Zoo supports.