diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt
index cdef369d15a..6ff87e5a3ff 100644
--- a/requirements-jax-cuda.txt
+++ b/requirements-jax-cuda.txt
@@ -1,5 +1,5 @@
 # Tensorflow cpu-only version (needed for testing).
-tf-nightly==2.15.0.dev20231009  # Pin a working nightly until rc0.
+tf-nightly-cpu==2.15.0.dev20231009  # Pin a working nightly until rc0.
 
 # Torch cpu-only version (needed for testing).
 --extra-index-url https://download.pytorch.org/whl/cpu
diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt
index 6418c2f558e..00e2b6457d2 100644
--- a/requirements-torch-cuda.txt
+++ b/requirements-torch-cuda.txt
@@ -1,5 +1,5 @@
 # Tensorflow cpu-only version (needed for testing).
-tf-nightly==2.15.0.dev20231009  # Pin a working nightly until rc0.
+tf-nightly-cpu==2.15.0.dev20231009  # Pin a working nightly until rc0.
 
 # Torch with cuda support.
 --extra-index-url https://download.pytorch.org/whl/cu118
diff --git a/requirements.txt b/requirements.txt
index 33eb38e34d9..ee83b8031c5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,8 @@
 # Tensorflow.
-tf-nightly==2.15.0.dev20231009  # Pin a working nightly until rc0.
+tf-nightly-cpu==2.15.0.dev20231009  # Pin a working nightly until rc0.
 
 # Torch.
+--extra-index-url https://download.pytorch.org/whl/cpu
 torch>=2.1.0
 torchvision>=0.16.0