File tree Expand file tree Collapse file tree 4 files changed +10
-10
lines changed Expand file tree Collapse file tree 4 files changed +10
-10
lines changed Original file line number Diff line number Diff line change 1
1
# Tensorflow cpu-only version (needed for testing).
2
- tensorflow-cpu~=2.18.0
2
+ tensorflow-cpu~=2.18.1
3
3
tf2onnx
4
4
5
5
# Torch cpu-only version (needed for testing).
6
6
--extra-index-url https://download.pytorch.org/whl/cpu
7
- torch==2.6.0+cpu
7
+ torch==2.6.0
8
8
9
9
# Jax with cuda support.
10
10
--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Original file line number Diff line number Diff line change 1
1
# Tensorflow with cuda support.
2
- tensorflow[and-cuda]~=2.18.0
2
+ tensorflow[and-cuda]~=2.18.1
3
3
tf2onnx
4
4
5
5
# Torch cpu-only version (needed for testing).
6
6
--extra-index-url https://download.pytorch.org/whl/cpu
7
- torch==2.6.0+cpu
7
+ torch==2.6.0
8
8
9
9
# Jax cpu-only version (needed for testing).
10
10
jax[cpu]
Original file line number Diff line number Diff line change 1
1
# Tensorflow cpu-only version (needed for testing).
2
- tensorflow-cpu~=2.18.0
2
+ tensorflow-cpu~=2.18.1
3
3
tf2onnx
4
4
5
5
# Torch with cuda support.
6
6
# - torch is pinned to a version that is compatible with torch-xla
7
7
# - torch-xla is pinned to a version that supports GPU (2.6 doesn't)
8
8
--extra-index-url https://download.pytorch.org/whl/cu121
9
- torch==2.5.1+cu121
10
- torch-xla==2.5.1 ;sys_platform != 'darwin'
9
+ torch==2.6.0
10
+ torch-xla==2.6.0 ;sys_platform != 'darwin'
11
11
12
12
# Jax cpu-only version (needed for testing).
13
13
jax[cpu]
Original file line number Diff line number Diff line change 1
1
# Tensorflow.
2
- tensorflow-cpu ~= 2.18.0 ;sys_platform != 'darwin'
3
- tensorflow ~= 2.18.0 ;sys_platform == 'darwin'
2
+ tensorflow-cpu ~= 2.18.1 ;sys_platform != 'darwin'
3
+ tensorflow ~= 2.18.1 ;sys_platform == 'darwin'
4
4
tf_keras
5
5
tf2onnx
6
6
7
7
# Torch.
8
8
--extra-index-url https://download.pytorch.org/whl/cpu
9
- torch == 2.6.0+cpu ;sys_platform != 'darwin'
9
+ torch == 2.6.0 ;sys_platform != 'darwin'
10
10
torch == 2.6.0 ;sys_platform == 'darwin'
11
11
torch-xla == 2.6.0 ;sys_platform != 'darwin'
12
12
You can’t perform that action at this time.
0 commit comments