pip install jax==0.4.2
pip install jaxlib==0.4.2+cuda11.cudnn86 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Last active
February 24, 2023 17:56
-
-
Save sivadasetty/749262e35a971cb1f6d26381b8f7f043 to your computer and use it in GitHub Desktop.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment