Facebook FAISS on Databricks

Been exploring this FAISS example on databricks, but I simply cannot get beyond indexing.

embeddings_dataset.add_faiss_index(column=“embeddings”)

leads to the following error.

ImportError: You must install Faiss to use FaissIndex. To do so you can run conda install -c pytorch faiss-cpu or conda install -c pytorch faiss-gpu. A community supported package is also available on pypi: pip install faiss-cpu or pip install faiss-gpu. Note that pip may not have the latest version of FAISS, and thus, some of the latest features and bug fixes may not be available.

No matter how I do it, I cannot manage to get FAISS working.