Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
73e5fb55
Unverified
Commit
73e5fb55
authored
Jun 14, 2019
by
Quan (Andy) Gan
Committed by
GitHub
Jun 14, 2019
Browse files
New installation guide for CUDA builds (#655)
parent
4be1d3e7
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
48 additions
and
2 deletions
+48
-2
docs/source/install/index.rst
docs/source/install/index.rst
+48
-2
No files found.
docs/source/install/index.rst
View file @
73e5fb55
...
@@ -21,6 +21,10 @@ DGL supports multiple tensor libraries (e.g. PyTorch, MXNet) as backends; refer
...
@@ -21,6 +21,10 @@ DGL supports multiple tensor libraries (e.g. PyTorch, MXNet) as backends; refer
`Working with different backends`_ for requirements on backends and how to select a
`Working with different backends`_ for requirements on backends and how to select a
backend.
backend.
Starting from 0.3 DGL is separated into CPU and CUDA builds. The builds share the
same Python package name, so installing DGL with CUDA 9 build after installing the
CPU build will overwrite the latter.
Install from conda
Install from conda
----------------------
----------------------
One can either grab `miniconda <https://conda.io/miniconda.html>`_ or
One can either grab `miniconda <https://conda.io/miniconda.html>`_ or
...
@@ -31,16 +35,58 @@ Once the conda environment is activated, run
...
@@ -31,16 +35,58 @@ Once the conda environment is activated, run
.. code:: bash
.. code:: bash
conda install -c dglteam dgl
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
Install from pip
Install from pip
----------------
----------------
O
ne can simply run the following command to install via ``pip``:
For CPU builds, o
ne can simply run the following command to install via ``pip``:
.. code:: bash
.. code:: bash
pip install dgl
pip install dgl
For CUDA builds, one needs to specify the URL:
.. code:: bash
pip install https://s3.us-east-2.amazonaws.com/dgl.ai/<BUILD-DIR>/dgl-0.3-<PYTHON-ABI>-<PLATFORM>.whl
where ``<BUILD-DIR>``, ``<PYTHON-ABI>`` and ``<PLATFORM>`` can take either of the following values:
+------------+---------------------+
| Build Type | ``<BUILD>`` |
+============+=====================+
| CUDA 9.0 | ``wheels/cuda9.0`` |
+------------+---------------------+
| CUDA 10.0 | ``wheels/cuda10.0`` |
+------------+---------------------+
+------------------+------------------+
| Python Version | ``<PYTHON-ABI>`` |
+==================+==================+
| Python 3.5 | ``cp35-cp35m`` |
+------------------+------------------+
| Python 3.6 | ``cp36-cp36m`` |
+------------------+------------------+
| Python 3.7 | ``cp37-cp37m`` |
+------------------+------------------+
+-------------+-----------------------+
| OS/Arch | ``<PLATFORM>`` |
+=============+=======================+
| Linux x64 | ``manylinux1_x86_64`` |
+-------------+-----------------------+
| Windows x64 | ``win_amd64`` |
+-------------+-----------------------+
For instance, if one wants to install the CUDA 9.0 build on Linux, Python 3.5, then the command is
.. code:: bash
pip install https://s3.us-east-2.amazonaws.com/dgl.ai/wheels/cuda9.0/dgl-0.3-cp35-cp35m-manylinux1_x86_64.whl
Working with different backends
Working with different backends
-------------------------------
-------------------------------
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment