diff --git a/README.md b/README.md
index 52fe5c45..3816eb60 100644
--- a/README.md
+++ b/README.md
@@ -47,7 +47,7 @@ typical install time ≈ 10 minutes.
 
 first install jax (with GPU support)
 
-``` pip install "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html ```
+``` pip install "jax[cuda]==0.7" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html ```
 
 second install colabdesign