Spark NLP Cheat Sheet

# Install Spark NLP from PyPI
$pip install spark-nlp==2.5.4

# Install Spark NLP from Anacodna/Conda
conda install -c johnsnowlabs spark-nlp

# Load Spark NLP with Spark Shell
spark-shell --packages com.johnsnowlabs.nlp:spark-nlp_2.11:2.5.4

# Load Spark NLP with PySpark
pyspark --packages com.johnsnowlabs.nlp:spark-nlp_2.11:2.5.4

# Load Spark NLP with Spark Submit
spark-submit --packages com.johnsnowlabs.nlp:spark-nlp_2.11:2.5.4

# Load Spark NLP as external JAR after comiling and bulding Spark NLP by `sbt assembly`
spark-shell --jar spark-nlp-assembly-2.5.4

NOTE: To use Spark NLP on Apache Spark 2.3.x you should instead use the following packages:

  • CPU: com.johnsnowlabs.nlp:spark-nlp-spark23_2.11:2.5.4
  • GPU: com.johnsnowlabs.nlp:spark-nlp-gpu-spark23_2.11:2.5.4


Quick Install

Let’s create a new Conda environment to manage all the dependencies there. You can use Python Virtual Environment if you prefer or not have any enviroment.

$ java -version
# should be Java 8 (Oracle or OpenJDK)
$ conda create -n sparknlp python=3.6 -y
$ conda activate sparknlp
$ pip install spark-nlp==2.5.4 pyspark==2.4.4

Of course you will need to have jupyter installed in your system:

pip install jupyter

Now you should be ready to create a jupyter notebook running from terminal:

jupyter notebook

Start Spark NLP Session from python

If you need to manually start SparkSession because you have other configuraations and sparknlp.start() is not including them, you can manually start the SparkSession:

spark = SparkSession.builder \
    .appName("Spark NLP")\
    .config("spark.driver.maxResultSize", "0") \
    .config("spark.jars.packages", "com.johnsnowlabs.nlp:spark-nlp_2.11:2.5.4")\
    .config("spark.kryoserializer.buffer.max", "1000M")\

Scala and Java

Our package is deployed to maven central. In order to add this package as a dependency in your application:

spark-nlp on Apacahe Spark 2.4.x:

<!-- -->


<!-- -->

spark-nlp on Apacahe Spark 2.3.x:

<!-- -->


<!-- -->


spark-nlp on Apacahe Spark 2.4.x:

libraryDependencies += "com.johnsnowlabs.nlp" %% "spark-nlp" % "2.5.4"


libraryDependencies += "com.johnsnowlabs.nlp" %% "spark-nlp-gpu" % "2.5.4"

spark-nlp on Apacahe Spark 2.3.x:

libraryDependencies += "com.johnsnowlabs.nlp" %% "spark-nlp-spark23" % "2.5.4"


libraryDependencies += "com.johnsnowlabs.nlp" %% "spark-nlp-gpu-spark23" % "2.5.4"

Maven Central:


Databricks Support

Spark NLP 2.5.4 has been tested and is compatible with the following runtimes:

  • 6.2
  • 6.2 ML
  • 6.3
  • 6.3 ML
  • 6.4
  • 6.4 ML
  • 6.5
  • 6.5 ML

Install Spark NLP on Databricks

  1. Create a cluster if you don’t have one already

  2. On a new cluster or existing one you need to add the following to the Aadvanced Options -> Spark tab:

spark.kryoserializer.buffer.max 1000M
spark.serializer org.apache.spark.serializer.KryoSerializer
  1. In Libraries tab inside your cluster you need to follow these steps:

    3.1. Insatll New -> PyPI -> spark-nlp -> Install

    3.2. Install New -> Maven -> Coordinates -> com.johnsnowlabs.nlp:spark-nlp_2.11:2.5.4 -> Install

  2. Now you can attach your notebook to the cluster and use Spark NLP!

Databricks Notebooks

You can view all the Databricks notebooks from this address:

Note: You can import these notebooks by using their URLs.

Last updated