Skip to content

Commit c46dc7d

Browse files
authored
fix(tf): fix UV resolution with TF 2.19 (#4786)
Fix #4679 <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **Chores** - Updated dependency metadata for TensorFlow 2.19.0 and TensorFlow-CPU 2.19.0 to improve compatibility and dependency resolution. <!-- end of auto-generated comment: release notes by coderabbit.ai --> Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
1 parent dacb5a2 commit c46dc7d

File tree

1 file changed

+83
-0
lines changed

1 file changed

+83
-0
lines changed

pyproject.toml

Lines changed: 83 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -467,3 +467,86 @@ openmpi = { index = "mpi4py" }
467467
name = "mpi4py"
468468
url = "https://pypi.anaconda.org/mpi4py/simple"
469469
explicit = true
470+
471+
[[tool.uv.dependency-metadata]]
472+
# Fix https://github.com/deepmodeling/deepmd-kit/issues/4679
473+
name = "tensorflow"
474+
version = "2.19.0"
475+
requires-dist = [
476+
'absl-py >=1.0.0',
477+
'astunparse >=1.6.0',
478+
'flatbuffers >=24.3.25',
479+
'gast !=0.5.0,!=0.5.1,!=0.5.2,>=0.2.1',
480+
'google-pasta >=0.1.1',
481+
'libclang >=13.0.0',
482+
'opt-einsum >=2.3.2',
483+
'packaging',
484+
'protobuf !=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.3',
485+
'requests <3,>=2.21.0',
486+
'setuptools',
487+
'six >=1.12.0',
488+
'termcolor >=1.1.0',
489+
'typing-extensions >=3.6.6',
490+
'wrapt >=1.11.0',
491+
'grpcio <2.0,>=1.24.3',
492+
'tensorboard ~=2.19.0',
493+
'keras >=3.5.0',
494+
'numpy <2.2.0,>=1.26.0',
495+
'h5py >=3.11.0',
496+
'ml-dtypes <1.0.0,>=0.5.1',
497+
# 'tensorflow-intel ==2.19.0 ; platform_system == "Windows"',
498+
'tensorflow-io-gcs-filesystem >=0.23.1 ; python_version < "3.12"',
499+
'nvidia-cublas-cu12 ==12.5.3.2 ; extra == "and-cuda"',
500+
'nvidia-cuda-cupti-cu12 ==12.5.82 ; extra == "and-cuda"',
501+
'nvidia-cuda-nvcc-cu12 ==12.5.82 ; extra == "and-cuda"',
502+
'nvidia-cuda-nvrtc-cu12 ==12.5.82 ; extra == "and-cuda"',
503+
'nvidia-cuda-runtime-cu12 ==12.5.82 ; extra == "and-cuda"',
504+
'nvidia-cudnn-cu12 ==9.3.0.75 ; extra == "and-cuda"',
505+
'nvidia-cufft-cu12 ==11.2.3.61 ; extra == "and-cuda"',
506+
'nvidia-curand-cu12 ==10.3.6.82 ; extra == "and-cuda"',
507+
'nvidia-cusolver-cu12 ==11.6.3.83 ; extra == "and-cuda"',
508+
'nvidia-cusparse-cu12 ==12.5.1.3 ; extra == "and-cuda"',
509+
'nvidia-nccl-cu12 ==2.23.4 ; extra == "and-cuda"',
510+
'nvidia-nvjitlink-cu12 ==12.5.82 ; extra == "and-cuda"',
511+
]
512+
513+
[[tool.uv.dependency-metadata]]
514+
name = "tensorflow-cpu"
515+
version = "2.19.0"
516+
requires-dist = [
517+
'absl-py >=1.0.0',
518+
'astunparse >=1.6.0',
519+
'flatbuffers >=24.3.25',
520+
'gast !=0.5.0,!=0.5.1,!=0.5.2,>=0.2.1',
521+
'google-pasta >=0.1.1',
522+
'libclang >=13.0.0',
523+
'opt-einsum >=2.3.2',
524+
'packaging',
525+
'protobuf !=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.3',
526+
'requests <3,>=2.21.0',
527+
'setuptools',
528+
'six >=1.12.0',
529+
'termcolor >=1.1.0',
530+
'typing-extensions >=3.6.6',
531+
'wrapt >=1.11.0',
532+
'grpcio <2.0,>=1.24.3',
533+
'tensorboard ~=2.19.0',
534+
'keras >=3.5.0',
535+
'numpy <2.2.0,>=1.26.0',
536+
'h5py >=3.11.0',
537+
'ml-dtypes <1.0.0,>=0.5.1',
538+
# 'tensorflow-intel ==2.19.0 ; platform_system == "Windows"',
539+
'tensorflow-io-gcs-filesystem >=0.23.1 ; python_version < "3.12"',
540+
'nvidia-cublas-cu12 ==12.5.3.2 ; extra == "and-cuda"',
541+
'nvidia-cuda-cupti-cu12 ==12.5.82 ; extra == "and-cuda"',
542+
'nvidia-cuda-nvcc-cu12 ==12.5.82 ; extra == "and-cuda"',
543+
'nvidia-cuda-nvrtc-cu12 ==12.5.82 ; extra == "and-cuda"',
544+
'nvidia-cuda-runtime-cu12 ==12.5.82 ; extra == "and-cuda"',
545+
'nvidia-cudnn-cu12 ==9.3.0.75 ; extra == "and-cuda"',
546+
'nvidia-cufft-cu12 ==11.2.3.61 ; extra == "and-cuda"',
547+
'nvidia-curand-cu12 ==10.3.6.82 ; extra == "and-cuda"',
548+
'nvidia-cusolver-cu12 ==11.6.3.83 ; extra == "and-cuda"',
549+
'nvidia-cusparse-cu12 ==12.5.1.3 ; extra == "and-cuda"',
550+
'nvidia-nccl-cu12 ==2.23.4 ; extra == "and-cuda"',
551+
'nvidia-nvjitlink-cu12 ==12.5.82 ; extra == "and-cuda"',
552+
]

0 commit comments

Comments
 (0)