Skip to content

Commit 5713afe

Browse files
authored
Update torch version to 2.9.0. (#165)
Both for CPU and CUDA.
1 parent 6887dcb commit 5713afe

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

requirements-torch-cuda.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ tensorflow-cpu
33

44
# Torch with cuda support.
55
--extra-index-url https://download.pytorch.org/whl/cu121
6-
torch==2.7.1
6+
torch==2.9.0
77

88
# Jax cpu-only version.
99
jax[cpu]

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ tensorflow>=2.20.0;sys_platform == 'darwin'
44

55
# Torch.
66
--extra-index-url https://download.pytorch.org/whl/cpu
7-
torch>=2.1.0
7+
torch>=2.9.0
88

99
# Jax.
1010
jax[cpu]>=0.7.0

0 commit comments

Comments
 (0)