项目文件夹

文件
2020-06-26 22:55:16 +08:00

199 行
5.5 KiB
ReStructuredText

Install DGL
===========
This topic explains how to install DGL. We recommend installing DGL by using ``conda`` or ``pip``.
System requirements
-------------------
DGL works with the following operating systems:
* Ubuntu 16.04
* macOS X
* Windows 10
DGL requires Python version 3.5 or later. Python 3.4 or earlier is not
tested.
DGL supports multiple tensor libraries as backends, e.g., PyTorch, MXNet. For requirements on backends and how to select one, see :ref:`backends`.
Starting at version 0.3, DGL is separated into CPU and CUDA builds. The builds share the
same Python package name. If you install DGL with a CUDA 9 build after you install the
CPU build, then the CPU build is overwritten.
Install from conda
----------------------
If ``conda`` is not yet installed, get either `miniconda <https://conda.io/miniconda.html>`_ or
the full `anaconda <https://www.anaconda.com/download/>`_.
With ``conda`` installed, you will want install DGL into Python 3.5 ``conda`` environment.
Run `conda create -n dgl python=3.5` to create the environment.
Activate the environment by running `source activate dgl`.
After the ``conda`` environment is activated, run one of the following commands.
.. code:: bash
conda install -c dglteam dgl # For CPU Build
conda install -c dglteam dgl-cuda9.0 # For CUDA 9.0 Build
conda install -c dglteam dgl-cuda10.0 # For CUDA 10.0 Build
conda install -c dglteam dgl-cuda10.1 # For CUDA 10.1 Build
conda install -c dglteam dgl-cuda10.2 # For CUDA 10.2 Build
Install from pip
----------------
For CPU builds, run the following command to install with ``pip``.
.. code:: bash
pip install dgl
For CUDA builds, run one of the following commands and specify the CUDA version.
.. code:: bash
pip install dgl # For CPU Build
pip install dgl-cu90 # For CUDA 9.0 Build
pip install dgl-cu92 # For CUDA 9.2 Build
pip install dgl-cu100 # For CUDA 10.0 Build
pip install dgl-cu101 # For CUDA 10.1 Build
For the most current nightly build from master branch, run one of the following commands.
.. code:: bash
pip install --pre dgl # For CPU Build
pip install --pre dgl-cu90 # For CUDA 9.0 Build
pip install --pre dgl-cu92 # For CUDA 9.2 Build
pip install --pre dgl-cu100 # For CUDA 10.0 Build
pip install --pre dgl-cu101 # For CUDA 10.1 Build
.. _install-from-source:
Install from source
-------------------
Download the source files from GitHub.
.. code:: bash
git clone --recurse-submodules https://github.com/dmlc/dgl.git
(Optional) Clone the repository first, and then run the following:
.. code:: bash
git submodule update --init --recursive
Linux
`````
Install the system packages for building the shared library. For Debian and Ubuntu
users, run:
.. code:: bash
sudo apt-get update
sudo apt-get install -y build-essential python3-dev make cmake
For Fedora/RHEL/CentOS users, run:
.. code:: bash
sudo yum install -y gcc-c++ python3-devel make cmake
Build the shared library. Use the configuration template ``cmake/config.cmake``.
Copy it to either the project directory or the build directory and change the
configuration as you wish. For example, change ``USE_CUDA`` to ``ON`` will
enable a CUDA build. You could also pass ``-DKEY=VALUE`` to the cmake command
for the same purpose.
- CPU-only build
.. code:: bash
mkdir build
cd build
cmake ..
make -j4
- CUDA build
.. code:: bash
mkdir build
cd build
cmake -DUSE_CUDA=ON ..
make -j4
Finally, install the Python binding.
.. code:: bash
cd ../python
python setup.py install
macOS
`````
Installation on macOS is similar to Linux. But macOS users need to install build tools like clang, GNU Make, and cmake first. These installation steps were tested on macOS X with clang 10.0.0, GNU Make 3.81, and cmake 3.13.1.
Tools like clang and GNU Make are packaged in **Command Line Tools** for macOS. To
install, run the following:
.. code:: bash
xcode-select --install
To install other needed packages like cmake, we recommend first installing
**Homebrew**, which is a popular package manager for macOS. To learn more, see the `Homebrew website <https://brew.sh/>`_.
After you install Homebrew, install cmake.
.. code:: bash
brew install cmake
Go to root directory of the DGL repository, build a shared library, and
install the Python binding for DGL.
.. code:: bash
mkdir build
cd build
cmake -DUSE_OPENMP=off ..
make -j4
cd ../python
python setup.py install
Windows
```````
The Windows source build is tested with CMake and MinGW/GCC. We highly recommend
using CMake and GCC from `conda installations <https://conda.io/miniconda.html>`_. To
get started, run the following:
.. code:: bash
conda install cmake m2w64-gcc m2w64-make
Build the shared library and install the Python binding.
.. code::
md build
cd build
cmake -DCMAKE_CXX_FLAGS="-DDMLC_LOG_STACK_TRACE=0 -DDGL_EXPORTS" -DCMAKE_MAKE_PROGRAM=mingw32-make .. -G "MSYS Makefiles"
mingw32-make
cd ..\python
python setup.py install
You can also build DGL with MSBuild. With `MS Build Tools <https://go.microsoft.com/fwlink/?linkid=840931>`_
and `CMake on Windows <https://cmake.org/download/>`_ installed, run the following
in VS2017 x64 Native tools command prompt.
.. code::
MD build
CD build
cmake -DCMAKE_CXX_FLAGS="/DDGL_EXPORTS" -DCMAKE_CONFIGURATION_TYPES="Release" .. -G "Visual Studio 15 2017 Win64"
msbuild dgl.sln
cd ..\python
python setup.py install