Unverified Commit da3b4c19 authored by Thomas Aarholt's avatar Thomas Aarholt Committed by GitHub
Browse files

[python-package] Improve readme on experimental cuda usage (#5386)

* Improve readme on experimental cuda usage

* Revert header style change
parent b7513245
...@@ -123,7 +123,9 @@ All requirements from `Build from Sources section <#build-from-sources>`__ apply ...@@ -123,7 +123,9 @@ All requirements from `Build from Sources section <#build-from-sources>`__ apply
**CUDA** library (version 9.0 or higher) is needed: details for installation can be found in `Installation Guide <https://github.com/microsoft/LightGBM/blob/master/docs/Installation-Guide.rst#build-cuda-version-experimental>`__. **CUDA** library (version 9.0 or higher) is needed: details for installation can be found in `Installation Guide <https://github.com/microsoft/LightGBM/blob/master/docs/Installation-Guide.rst#build-cuda-version-experimental>`__.
Recently, a new CUDA version with better efficiency is implemented as an experimental feature. To build the new CUDA version, replace ``--cuda`` with ``--cuda-exp`` in the above commands. Please note that new version requires **CUDA** 10.0 or later libraries. Recently, a new CUDA version with better efficiency is implemented as an experimental feature. To build the new CUDA version, replace ``--cuda`` with ``--cuda-exp`` in the above commands. Please note that new version requires **CUDA** 10.0 or later libraries. Note that this new version uses twice the memory, since it stores data row-wise as well as column-wise in memory to improve performance (see this `issue <https://github.com/microsoft/LightGBM/issues/5318>`__ for discussion).
To use the regular or experimental cuda versions within Python, pass ``{"device":"cuda"}`` or ``{"device":"cuda_exp"}`` respectively as parameters.
Build HDFS Version Build HDFS Version
~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment