-
Notifications
You must be signed in to change notification settings - Fork 2.7k
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
Jax metal failed to install #19886
Comments
Based on the packages, it is AMD GPU? Could you try a venv with python=3.9? |
Reproduces on my m2 mac. with both py 3.10.6 and 3.9.13 |
Tried jax==0.4.11 jaxlib==0.4.11 jax-metal==0.0.4 - same thing |
Haven't been able to reproduce the issue. The below config shows an installation and verification result:
|
Right, i think i was able to figure it out - in my case it was due python being |
I have just tried to install following the instructions in the apple website (https://developer.apple.com/metal/jax/) and it failed. Same error than everyone here in a M2. How did you switched your native python3? I have just ran the following code: import platform
# Check the machine architecture
machine = platform.machine()
if machine == 'arm64':
print("Your Python version is ARM64")
elif machine == 'i386':
print("Your Python version is i386 (32-bit)")
elif machine == 'x86_64':
print("Your Python version is x86_64 (64-bit)")
else:
print(f"Unknown machine architecture: {machine}") and the print out is:
|
@phisanti you switch in you CLI with |
@curlup thanks for the tip. It worked for me! |
Description
Using the instructions on the pip website the jax_metal failed to install
System info (python version, jaxlib version, accelerator, etc.)
Macbook Air M2
Macos Sonoma 14.3.1 (23D60)
Python 3.10
The text was updated successfully, but these errors were encountered: