Skip to content

Different cuda versions specified in pip install command between github and docs #31907

@Nebulised

Description

@Nebulised

Description

Minor documentation discrepancy?

Noticed the github page installation instructions has

pip install -U "jax[cuda13]"

The jax documentation site has

pip install -U "jax[cuda12]"

with further down the uncommented upgrade version being CUDA13

pip install --upgrade "jax[cuda13]"

System info (python version, jaxlib version, accelerator, etc.)

N/A

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions