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

Unable to install jaxlib on Windows using bazel #13617

Closed
keepingupwithdeeplearning opened this issue Jun 26, 2021 · 3 comments
Closed

Unable to install jaxlib on Windows using bazel #13617

keepingupwithdeeplearning opened this issue Jun 26, 2021 · 3 comments
Labels
area-Windows Windows-specific issues and feature requests more data needed

Comments

@keepingupwithdeeplearning
Copy link

keepingupwithdeeplearning commented Jun 26, 2021

I have been trying to install jaxlib on Windows10 using bazel 4.0.0 and msys2. But it keeps giving me the following cuda related error:
Screenshot (54)

If I use bazel 4.1.0, I get the following cpu related error:

Screenshot (55)

@aiuto aiuto added area-Windows Windows-specific issues and feature requests more data needed labels Jun 27, 2021
@aiuto
Copy link
Contributor

aiuto commented Jun 27, 2021

Can you provide a short set of steps to reproduce this issue.

@keepingupwithdeeplearning
Copy link
Author

keepingupwithdeeplearning commented Jun 28, 2021

I'm building jaxlib on Windows10 from source:
https://jax.readthedocs.io/en/latest/developer.html#building-jaxlib-from-source

After downloading jaxlib source code and installing bazel 4.0.0 and msys2, I used the following command :

python .\build\build.py --enable_cuda --cuda_path="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1" --cudnn_path="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1" --cuda_compute_capabilities="6.1" --cuda_version="10.1" --cudnn_version="7.6.5"

Thanks.

@aiuto
Copy link
Contributor

aiuto commented Jun 28, 2021

You should ask this question wherever jaxlib go for support. Someone would have to trace down their build.py script and understand that to make any useful comments. That is not something the Bazel team would know about.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
area-Windows Windows-specific issues and feature requests more data needed
Projects
None yet
Development

No branches or pull requests

2 participants