Install Graph4NLP

Currently, users can install Graph4NLP via pip or source code. Graph4NLP supports the following OSes:

  • Linux-based systems (tested on Ubuntu 18.04 and later)

  • macOS (only CPU version)

  • Windows 10 (only support pytorch >= 1.8)

Installation via pip (binaries)

We provide pip wheels for all major OS/PyTorch/CUDA combinations. Note that we highly recommend Windows users refer to Installation via source code due to compatibility.

Ensure that at least PyTorch (>=1.6.0) is installed:

Note that >=1.6.0 is ok.

$ python -c "import torch; print(torch.__version__)"
>>> 1.6.0

Find the CUDA version PyTorch was installed with (for GPU users):

$ python -c "import torch; print(torch.version.cuda)"
>>> 10.2

Install the relevant dependencies:

torchtext is needed since Graph4NLP relies on it to implement embeddings. Please pay attention to the PyTorch requirements before installing torchtext with the following script! For detailed version matching please refer [here](https://pypi.org/project/torchtext/).

Install Graph4NLP

pip install graph4nlp${CUDA}

where ${CUDA} should be replaced by the specific CUDA version (none (CPU version), “-cu92”, “-cu101”, “-cu102”, “-cu110”). The following table shows the concrete command lines. For CUDA 11.1 users, please refer to Installation via source code.

Supported platforms

Platform

Command

CPU

pip install graph4nlp

CUDA 9.2

pip install graph4nlp-cu92

CUDA 10.1

pip install graph4nlp-cu101

CUDA 10.2

pip install graph4nlp-cu102

CUDA 11.0

pip install graph4nlp-cu110

Installation via source code

Ensure that at least PyTorch (>=1.6.0) is installed:

Note that >=1.6.0 is ok.

$ python -c "import torch; print(torch.__version__)"
>>> 1.6.0

Find the CUDA version PyTorch was installed with (for GPU users):

$ python -c "import torch; print(torch.version.cuda)"
>>> 10.2

Install the relevant dependencies:

torchtext is needed since Graph4NLP relies on it to implement embeddings. Please pay attention to the PyTorch requirements before installing torchtext with the following script! For detailed version matching please refer [here](https://pypi.org/project/torchtext/).

Download the source code of Graph4NLP from Github:

git clone https://github.com/graph4ai/graph4nlp.git
cd graph4nlp

Configure the CUDA version

Then run ./configure (or ./configure.bat if you are using Windows 10) to config your installation. The configuration program will ask you to specify your CUDA version. If you do not have a GPU, please type ‘cpu’.

./configure

Install Graph4NLP

Finally, install the package:

python setup.py install

Enjoy!