mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Note that the MSVC studio 2019 redistributable is required for JAX on Windows.

Browse Source 
Issue #16664
This commit is contained in:
Peter Hawkins 2023-07-11 12:45:13 -04:00
parent e894e4817a
commit a1a5159fbf
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
README.md
View File

@ -401,6 +401,13 @@ they can use the *experimental* native Windows CPU-only support.
### pip installation: CPU ### pip installation: CPU
We currently release `jaxlib` wheels for the following
operating systems and architectures:
* Linux, x86-64
* Mac, Intel
* Mac, ARM
* Windows, x86-64 (*experimental*)
To install a CPU-only version of JAX, which might be useful for doing local To install a CPU-only version of JAX, which might be useful for doing local
development on a laptop, you can run development on a laptop, you can run
@ -409,13 +416,9 @@ pip install --upgrade pip
pip install --upgrade "jax[cpu]" pip install --upgrade "jax[cpu]"
``` ```
On Linux, it is often necessary to first update `pip` to a version that supports On Windows, you may also need to install the
`manylinux2014` wheels. We currently release `jaxlib` wheels for the following [Microsoft Visual Studio 2019 Redistributable](https://learn.microsoft.com/en-US/cpp/windows/latest-supported-vc-redist?view=msvc-170#visual-studio-2015-2017-2019-and-2022)
operating systems and architectures: if it is not already installed on your machine.
* Linux, x86-64
* Mac, Intel
* Mac, ARM
* Windows, x86-64 (*experimental*)
Other operating systems and architectures require building from source. Trying Other operating systems and architectures require building from source. Trying
to pip install on other operating systems and architectures may lead to `jaxlib` to pip install on other operating systems and architectures may lead to `jaxlib`