use cpu for backends when not testing on gpu (#18661)

keras-team · Oct 23, 2023 · 742a2d0 · 742a2d0
1 parent 2ad8e07
commit 742a2d0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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