lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pin torch < 2.3.0 (#19603)

Browse Source
This commit is contained in:
Ramesh Sampath 2024-04-23 15:36:35 -05:00 committed by GitHub
parent c61857d380
commit d84d5ccd06
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements-jax-cuda.txt
View File

@ -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.

2
requirements-tensorflow-cuda.txt
View File

@ -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).

3
requirements.txt
View File

@ -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.