How to install a specific version of jax[cuda12] #29644
Answered
by
jakevdp
YousefElbrolosy
asked this question in
Q&A
-
|
I want to install jax[cuda12]==0.5.2, because I am using the version of jax 0.5.2 and I want to use the gpu compatible version with that, however I cannot seem to be able to specify the exact version of jax[cuda12] I want, could someone please direct me to how I can restrict the version? |
Beta Was this translation helpful? Give feedback.
Answered by
jakevdp
Jun 23, 2025
Replies: 1 comment 1 reply
-
|
Have you tried something like |
Beta Was this translation helpful? Give feedback.
1 reply
-
|
Yes this worked thank you! |
Beta Was this translation helpful? Give feedback.
Answer selected by
YousefElbrolosy
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Have you tried something like
pip install -U "jax[cuda12]==0.5.2"?