Merge pull request #28 from vwxyzjn/fix-readme

Fix instruction for installing cuda jax
ikostrikov · Jun 25, 2022 · 3674862 · 3674862
2 parents 6550d6f + 3477d0c
commit 3674862
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -70,7 +70,7 @@ To install, run
 ```bash
 poetry install
 # For GPU support run
-pip install "jax[cuda]==0.3.10" -f https://storage.googleapis.com/jax-releases/jax_releases.html
+pip install "jax[cuda]==0.3.10" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 ```
 
 For further instructions on running this code on GPU, please follow instructions from [the official repository](https://github.com/google/jax).