Skip to content

Commit

Permalink
docs(python): Update GPU engine installation instructions to remove `…
Browse files Browse the repository at this point in the history
…--extra-index-url` from CUDA 12 packages (#20381)

Co-authored-by: Bradley Dice <[email protected]>
  • Loading branch information
beckernick and bdice authored Dec 20, 2024
1 parent 2c9e503 commit c3a03cc
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 6 deletions.
4 changes: 2 additions & 2 deletions docs/source/user-guide/gpu-support.md
Original file line number Diff line number Diff line change
Expand Up @@ -20,12 +20,12 @@ You can install the GPU backend for Polars with a feature flag as part of a norm
=== ":fontawesome-brands-python: Python"

```bash
pip install --extra-index-url=https://pypi.nvidia.com polars[gpu]
pip install polars[gpu]
```

!!! note Installation on a CUDA 11 system

If you have CUDA 11, the installation line is slightly more complicated: the relevant GPU package must be requested by hand.
If you have CUDA 11, the installation line also needs the NVIDIA package index to get the CUDA 11 package.

=== ":fontawesome-brands-python: Python"
```bash
Expand Down
4 changes: 1 addition & 3 deletions docs/source/user-guide/installation.md
Original file line number Diff line number Diff line change
Expand Up @@ -98,9 +98,7 @@ pip install 'polars[numpy,fsspec]'

!!! note

To install the GPU engine, you need to pass
`--extra-index-url=https://pypi.nvidia.com` to `pip`. See [GPU
support](gpu-support.md) for more detailed instructions and
See [GPU support](gpu-support.md) for more detailed instructions and
prerequisites.

#### Interoperability
Expand Down
2 changes: 1 addition & 1 deletion py-polars/polars/lazyframe/frame.py
Original file line number Diff line number Diff line change
Expand Up @@ -2019,7 +2019,7 @@ def collect(
err_prefix="GPU engine requested, but required package",
install_message=(
"Please install using the command "
"`pip install --extra-index-url=https://pypi.nvidia.com cudf-polars-cu12` "
"`pip install cudf-polars-cu12` "
"(or `pip install --extra-index-url=https://pypi.nvidia.com cudf-polars-cu11` "
"if your system has a CUDA 11 driver)."
),
Expand Down

0 comments on commit c3a03cc

Please sign in to comment.