Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add support for Jax on Windows #438

Open
dwang55 opened this issue Feb 23, 2019 · 6 comments

Comments

Projects
None yet
5 participants
@dwang55
Copy link

commented Feb 23, 2019

I failed to build jax from source with(similar with #253, but my system is win):
Traceback (most recent call last):
File "build/build.py", line 308, in
main()
File "build/build.py", line 304, in main
[":install_xla_in_source_tree", os.getcwd()])
File "build/build.py", line 50, in shell
output = subprocess.check_output(cmd)
File "C:\Users\wangd\AppData\Local\Programs\Python\Python36\lib\subprocess.py", line 336, in check_output
**kwargs).stdout
File "C:\Users\wangd\AppData\Local\Programs\Python\Python36\lib\subprocess.py", line 418, in run
output=stdout, stderr=stderr)
subprocess.CalledProcessError: Command '['C:\bazel\bazel.EXE', 'run', '--verbose_failures=true', '--config=mkl_open_source_only', ':install_xla_in_source_tree', 'C:\Users\wangd\jax\build']' returned non-zero exit status 2.

@mattjj

This comment has been minimized.

Copy link
Collaborator

commented Feb 23, 2019

Thanks for raising this.

JAX doesn’t support Windows right now, though it may be that the only thing standing in our way is getting the build to work. If Tensorflow can be built with XLA on Windows, then it should be possible to build JAX too, with some revisions to our own build scripts to fix issues like this.

Getting JAX on Windows is out of scope for us for now, just because we need to prioritize other kinds of progress. But contributions are welcome!

@mattjj mattjj closed this Feb 23, 2019

@mattjj mattjj reopened this Feb 23, 2019

@mattjj

This comment has been minimized.

Copy link
Collaborator

commented Feb 23, 2019

On second thought, let’s leave this issue open with a “contributions welcome” tag so that it’s visible to others.

@mattjj mattjj added good first issue and removed wontfix labels Feb 23, 2019

@dwang55

This comment has been minimized.

Copy link
Author

commented Feb 24, 2019

Thanks for the response! I'm following the instruction from Eric Jang's MAML tutorial. Maybe it's helpful to illustrate it in README for beginners on machine learning like me.

@hawkinsp hawkinsp changed the title Failed to build jax from source on Windows Add support for Jax on Windows Mar 14, 2019

@Prasitt

This comment has been minimized.

Copy link

commented Mar 22, 2019

7.2

@cshesse

This comment has been minimized.

Copy link

commented Apr 9, 2019

Looks like XLA does not compile on Windows as of version v2.0.0-alpha and v1.13.1: tensorflow/tensorflow#25773

@cshesse

This comment has been minimized.

Copy link

commented May 7, 2019

That issue has been fixed, so it's possible that jax compiles on windows now. In the meantime, I installed the CPU only version of jax using Windows Subsystem for Linux and it seems to work fine there.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.