diff --git a/.github/workflows/package_c.yml b/.github/workflows/package_c.yml index f5e4a97d56..4ffd6829fc 100644 --- a/.github/workflows/package_c.yml +++ b/.github/workflows/package_c.yml @@ -22,7 +22,7 @@ jobs: tensorflow_version: "" filename: libdeepmd_c.tar.gz - tensorflow_build_version: "2.14" - tensorflow_version: ">=2.5.0rc0,<2.15" + tensorflow_version: ">=2.5.0,<2.15" filename: libdeepmd_c_cu11.tar.gz steps: - uses: actions/checkout@v4 diff --git a/backend/find_tensorflow.py b/backend/find_tensorflow.py index 1fc3a8a6d9..8b055f9f4f 100644 --- a/backend/find_tensorflow.py +++ b/backend/find_tensorflow.py @@ -88,14 +88,14 @@ def find_tensorflow() -> tuple[Optional[str], list[str]]: # CUDA 12.2, cudnn 9 requires.extend( [ - "tensorflow-cpu>=2.18.0rc0; platform_machine=='x86_64' and platform_system == 'Linux'", + "tensorflow-cpu>=2.18.0; platform_machine=='x86_64' and platform_system == 'Linux'", ] ) elif cuda_version in SpecifierSet(">=11,<12"): # CUDA 11.8, cudnn 8 requires.extend( [ - "tensorflow-cpu>=2.5.0rc0,<2.15; platform_machine=='x86_64' and platform_system == 'Linux'", + "tensorflow-cpu>=2.5.0,<2.15; platform_machine=='x86_64' and platform_system == 'Linux'", ] ) tf_version = "2.14.1" diff --git a/pyproject.toml b/pyproject.toml index 8af6b85cdb..feb790527b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -90,7 +90,7 @@ test = [ docs = [ "sphinx>=3.1.1", "sphinx-book-theme", - "myst-nb>=1.0.0rc0", + "myst-nb>=1.0.0", "myst-parser>=0.19.2", "sphinx-design", "breathe",