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
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
development on a laptop, you can run
@ -409,13 +416,9 @@ pip install --upgrade pip
pip install --upgrade "jax[cpu]"
```
On Linux, it is often necessary to first update `pip` to a version that supports
`manylinux2014` wheels. We currently release `jaxlib` wheels for the following
operating systems and architectures:
* Linux, x86-64
* Mac, Intel
* Mac, ARM
* Windows, x86-64 (*experimental*)
On Windows, you may also need to install the
[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)
if it is not already installed on your machine.
Other operating systems and architectures require building from source. Trying
to pip install on other operating systems and architectures may lead to `jaxlib`