We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 94b0a55 commit 97b0254Copy full SHA for 97b0254
requirements-torch-cuda.txt
@@ -3,7 +3,7 @@ tensorflow-cpu>=2.16.1
3
4
# Torch with cuda support.
5
--extra-index-url https://download.pytorch.org/whl/cu121
6
-torch==2.4.0+cu121
+torch==2.6.0
7
torchvision==0.19.0+cu121
8
9
# Jax cpu-only version.
0 commit comments