Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[doc] Install the conda GPU variant in environments without CUDA (#10731)

Browse Source
This commit is contained in:
James Lamb 2024-08-22 06:48:15 -05:00 committed by GitHub
parent cd83fe6033
commit dbfafd8557
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
doc/install.rst
View File

@ -107,6 +107,13 @@ Conda should be able to detect the existence of a GPU on your machine and instal
# Use NVIDIA GPU # Use NVIDIA GPU
conda install -c conda-forge py-xgboost-gpu conda install -c conda-forge py-xgboost-gpu
To force the installation of the GPU variant on a machine that does not have an NVIDIA GPU, use environment variable ``CONDA_OVERRIDE_CUDA``,
as described in `"Managing Virtual Packages" in the conda docs <https://conda.io/projects/conda/en/latest/user-guide/tasks/manage-virtual.html>`_.
.. code-block:: bash
export CONDA_OVERRIDE_CUDA="12.5"
conda install -c conda-forge py-xgboost-gpu
Visit the `Miniconda website <https://docs.conda.io/en/latest/miniconda.html>`_ to obtain Conda. Visit the `Miniconda website <https://docs.conda.io/en/latest/miniconda.html>`_ to obtain Conda.