From 4a5a77f7fac93457ee064ce7064eac6a84f5ec39 Mon Sep 17 00:00:00 2001 From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com> Date: Mon, 15 Dec 2025 22:49:09 +0000 Subject: [PATCH 1/3] build(deps-dev): bump jax from 0.5.0 to 0.6.2 Bumps [jax](https://github.com/jax-ml/jax) from 0.5.0 to 0.6.2. - [Release notes](https://github.com/jax-ml/jax/releases) - [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md) - [Commits](https://github.com/jax-ml/jax/compare/jax-v0.5.0...jax-v0.6.2) --- updated-dependencies: - dependency-name: jax dependency-version: 0.6.2 dependency-type: direct:development update-type: version-update:semver-minor ... Signed-off-by: dependabot[bot] --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 5ad5101c01..1e5696896e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -169,7 +169,7 @@ pin_pytorch_gpu = [ "torch>=2.7,<2.10", ] pin_jax = [ - "jax==0.5.0;python_version>='3.10'", + "jax==0.6.2;python_version>='3.10'", ] [tool.setuptools_scm] From a43ab8029c40437fbb1d7d542464f9d9fbdc74bc Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Thu, 18 Dec 2025 01:30:28 +0800 Subject: [PATCH 2/3] bump cuda to 12.9.1 Signed-off-by: Jinzhe Zeng --- .github/workflows/test_cuda.yml | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml index a934401a20..d95a440f55 100644 --- a/.github/workflows/test_cuda.yml +++ b/.github/workflows/test_cuda.yml @@ -19,7 +19,7 @@ jobs: runs-on: nvidia # https://github.com/deepmodeling/deepmd-kit/pull/2884#issuecomment-1744216845 container: - image: nvidia/cuda:12.6.2-cudnn-devel-ubuntu22.04 + image: nvidia/cuda:12.9.1-cudnn-devel-ubuntu22.04 options: --gpus all if: github.repository_owner == 'deepmodeling' && (github.event_name == 'pull_request' && github.event.label && github.event.label.name == 'Test CUDA' || github.event_name == 'workflow_dispatch' || github.event_name == 'merge_group') steps: @@ -36,12 +36,6 @@ jobs: with: useLocalCache: true useCloudCache: false - - run: | - wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb \ - && sudo dpkg -i cuda-keyring_1.0-1_all.deb \ - && sudo apt-get update \ - && sudo apt-get -y install cuda-12-3 libcudnn8=8.9.5.*-1+cuda12.3 - if: false # skip as we use nvidia image - run: python -m pip install -U uv - run: source/install/uv_with_retry.sh pip install --system --group pin_tensorflow_gpu --group pin_pytorch_gpu --group pin_jax "jax[cuda12]" - run: | From c988731117cc5200814e17ac8e391d01eaea144a Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Thu, 25 Dec 2025 01:54:14 +0800 Subject: [PATCH 3/3] try TF 2.19.1 --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 1e5696896e..7be0554070 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -157,7 +157,7 @@ pin_tensorflow_cpu = [ "tensorflow; platform_machine=='x86_64' and platform_system == 'Darwin'", ] pin_tensorflow_gpu = [ - "tensorflow~=2.18.0", + "tensorflow==2.19.1", ] pin_pytorch_cpu = [ # https://github.com/pytorch/pytorch/issues/114602