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.
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