pin torch < 2.3.0 (#19603)
This commit is contained in:
parent
c61857d380
commit
d84d5ccd06
@ -3,7 +3,7 @@ tensorflow-cpu~=2.16.1 # Pin to TF 2.16
|
|||||||
|
|
||||||
# Torch cpu-only version (needed for testing).
|
# Torch cpu-only version (needed for testing).
|
||||||
--extra-index-url https://download.pytorch.org/whl/cpu
|
--extra-index-url https://download.pytorch.org/whl/cpu
|
||||||
torch>=2.1.0
|
torch>=2.1.0, <2.3.0
|
||||||
torchvision>=0.16.0
|
torchvision>=0.16.0
|
||||||
|
|
||||||
# Jax with cuda support.
|
# Jax with cuda support.
|
||||||
|
@ -3,7 +3,7 @@ tensorflow[and-cuda]~=2.16.1 # Pin to TF 2.16
|
|||||||
|
|
||||||
# Torch cpu-only version (needed for testing).
|
# Torch cpu-only version (needed for testing).
|
||||||
--extra-index-url https://download.pytorch.org/whl/cpu
|
--extra-index-url https://download.pytorch.org/whl/cpu
|
||||||
torch>=2.1.0
|
torch>=2.1.0, <2.3.0
|
||||||
torchvision>=0.16.0
|
torchvision>=0.16.0
|
||||||
|
|
||||||
# Jax cpu-only version (needed for testing).
|
# Jax cpu-only version (needed for testing).
|
||||||
|
@ -2,8 +2,9 @@
|
|||||||
tensorflow-cpu~=2.16.1 # Pin to TF 2.16
|
tensorflow-cpu~=2.16.1 # Pin to TF 2.16
|
||||||
|
|
||||||
# Torch.
|
# Torch.
|
||||||
|
# TODO: Pin to < 2.3.0 (GitHub issue #19602)
|
||||||
--extra-index-url https://download.pytorch.org/whl/cpu
|
--extra-index-url https://download.pytorch.org/whl/cpu
|
||||||
torch>=2.1.0
|
torch>=2.1.0, <2.3.0
|
||||||
torchvision>=0.16.0
|
torchvision>=0.16.0
|
||||||
|
|
||||||
# Jax.
|
# Jax.
|
||||||
|
Loading…
Reference in New Issue
Block a user