From 7d792e6e03ad0da659e1d554316a7ef28b3cc405 Mon Sep 17 00:00:00 2001 From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com> Date: Thu, 2 Apr 2026 10:55:29 +0000 Subject: [PATCH 1/3] build(deps-dev): bump torch from 2.10.0 to 2.11.0 Bumps [torch](https://github.com/pytorch/pytorch) from 2.10.0 to 2.11.0. - [Release notes](https://github.com/pytorch/pytorch/releases) - [Changelog](https://github.com/pytorch/pytorch/blob/main/RELEASE.md) - [Commits](https://github.com/pytorch/pytorch/compare/v2.10.0...v2.11.0) --- updated-dependencies: - dependency-name: torch dependency-version: 2.11.0 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 2f8f86a713..ea33883fe2 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -170,7 +170,7 @@ pin_pytorch_cpu = [ "torch; platform_machine=='x86_64' and platform_system == 'Darwin'", ] pin_pytorch_gpu = [ - "torch==2.10.0", + "torch==2.11.0", ] pin_jax = [ "jax==0.5.0;python_version>='3.10'", From 834f97eff5cc72f74f16621f5ac3dd0256f04f88 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Fri, 3 Apr 2026 22:21:18 +0800 Subject: [PATCH 2/3] set torch_backend to cu128 Signed-off-by: Jinzhe Zeng --- .github/workflows/test_cuda.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml index 65773ccbfe..3efcd079ea 100644 --- a/.github/workflows/test_cuda.yml +++ b/.github/workflows/test_cuda.yml @@ -43,7 +43,7 @@ jobs: && 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: source/install/uv_with_retry.sh pip install --system --group pin_tensorflow_gpu --group pin_pytorch_gpu --group pin_jax "jax[cuda12]" --torch-backend cu128 - run: | export PYTORCH_ROOT=$(python -c 'import torch;print(torch.__path__[0])') export TENSORFLOW_ROOT=$(python -c 'import importlib.util,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)') From 69a68baec6fa707f364422d22bb3ee1308f6f923 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Sat, 4 Apr 2026 22:09:41 +0800 Subject: [PATCH 3/3] Add nvidia-nccl-cu12 dependency to test_cuda.yml Signed-off-by: Jinzhe Zeng --- .github/workflows/test_cuda.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml index 3efcd079ea..855029963e 100644 --- a/.github/workflows/test_cuda.yml +++ b/.github/workflows/test_cuda.yml @@ -50,7 +50,7 @@ jobs: pip install --find-links "https://www.paddlepaddle.org.cn/packages/nightly/cu126/paddlepaddle-gpu/" --index-url https://pypi.org/simple --trusted-host www.paddlepaddle.org.cn --trusted-host paddlepaddle.org.cn "paddlepaddle-gpu==3.4.0.dev20260310" source/install/uv_with_retry.sh pip install --system -v -e .[gpu,test,lmp,cu12,torch,jax] mpi4py --reinstall-package deepmd-kit # See https://github.com/jax-ml/jax/issues/29042 - source/install/uv_with_retry.sh pip install --system -U 'nvidia-cublas-cu12>=12.9.0.13' + source/install/uv_with_retry.sh pip install --system -U 'nvidia-cublas-cu12>=12.9.0.13' 'nvidia-nccl-cu12>=2.28.9' env: DP_VARIANT: cuda DP_ENABLE_NATIVE_OPTIMIZATION: 1