Skip to content

Commit

Permalink
Add x86_64 dependency note to pip installation
Browse files Browse the repository at this point in the history
Currently non x86_64 linux architectures are not supported, see jax-ml#7097 for request to change this. This can lead to installation confusion, as jax will install, but jaxlib will not. For example see jax-ml#12307. 

If this can be more clearly phrased or explained, let me know.
  • Loading branch information
treyra authored Oct 27, 2022
1 parent 66e75ed commit 96bd0f7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -392,7 +392,7 @@ binary package with `pip` or `conda`, or to [build JAX from
source](https://jax.readthedocs.io/en/latest/developer.html#building-from-source).

We support installing or building `jaxlib` on Linux (Ubuntu 16.04 or later) and
macOS (10.12 or later) platforms.
macOS (10.12 or later) platforms.

Windows users can use JAX on CPU and GPU via the [Windows Subsystem for
Linux](https://docs.microsoft.com/en-us/windows/wsl/about). In addition, there
Expand All @@ -413,7 +413,7 @@ pip install --upgrade "jax[cpu]"
```

On Linux, it is often necessary to first update `pip` to a version that supports
`manylinux2014` wheels.
`manylinux2014` wheels. Note these existing wheels are currently for `x86_64` architectures only. Trying to pip install with other architectures may lead to `jaxlib` not being installed alongside `jax`, although `jax` may successfully install.
**These `pip` installations do not work with Windows, and may fail silently; see
[above](#installation).**

Expand Down

0 comments on commit 96bd0f7

Please sign in to comment.