Align keras requirments fo torch (#2272)

sampathweb · web-flow · commit 10504cb00ed1 · 2024-01-05T06:52:04.000+05:30
diff --git a/requirements-tensorflow-cuda.txt b/requirements-tensorflow-cuda.txt
@@ -1,5 +1,4 @@
 # Tensorflow with cuda support.
---extra-index-url https://pypi.nvidia.com
 tf-nightly[and-cuda]==2.16.0.dev20231221  # Pin a working nightly until rc0.
 
 # Torch cpu-only version.
diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt
@@ -2,9 +2,9 @@
 tf-nightly-cpu==2.16.0.dev20231221  # Pin a working nightly until rc0.
 
 # Torch with cuda support.
---extra-index-url https://download.pytorch.org/whl/cu118
-torch==2.1.2+cu118
-torchvision==0.16.2+cu118
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.1.2
+torchvision==0.16.2
 
 # Jax cpu-only version.
 jax[cpu]

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,4 @@`
`1`	`1`	`# Tensorflow with cuda support.`
`2`		`---extra-index-url https://pypi.nvidia.com`
`3`	`2`	`tf-nightly[and-cuda]==2.16.0.dev20231221 # Pin a working nightly until rc0.`
`4`	`3`
`5`	`4`	`# Torch cpu-only version.`