There are 2 ways to install GraphVite.

Install from conda

You can install GraphVite from conda with only one line.

conda install -c milagraph graphvite cudatoolkit=x.x

where x.x is your CUDA version, e.g. 9.2 or 10.0.

By default, this will install all dependencies, including PyTorch and matplotlib. If you only need embedding training without evaluation, you can take the following alternative with minimum dependencies.

conda install -c milagraph graphvite-mini cudatoolkit=x.x

Install from source

First, clone GraphVite from GitHub.

git clone
cd graphvite

Install compilation and runtime dependencies via conda.

conda install -y --file conda/requirements.txt

Compile the code using the following directives. If you have faiss installed from source, you can pass -DFAISS_PATH=/path/to/faiss to cmake.

mkdir build
cd build && cmake .. && make && cd -

Finally, install Python bindings.

cd python && python install && cd -