heavyai

This package enables using common Python data science toolkits with HeavyDB. It brings data frame support on CPU and GPU as well as support for arrow. See the documentation for more.

Quick Install (CPU)

Packages are available on conda-forge and PyPI:

# using conda-forge
conda install -c conda-forge heavyai

# using pip
pip install heavyai

Quick Install (GPU)

We recommend creating a fresh conda 3.8 or 3.9 environment when installing heavyai with GPU capabilities.

To install heavyai for GPU Dataframe support (conda-only):

conda create -n heavyai-gpu -c rapidsai -c nvidia -c conda-forge -c defaults python cudf cudatoolkit heavyai

Note that pyheavydb needs to be installed in the environment with pip until heavydb is available on conda-forge.

conda activate heavyai-gpu
pip install pyheavydb

Documentation

Further documentation for heavyai usage is available at: http://heavyai.readthedocs.io/