From c86f30f7b7afc85b97930de6f79f37ea9d7f4205 Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Wed, 11 Mar 2026 13:52:45 +0800 Subject: [PATCH 1/2] bump paddlepaddle version to 3.3.0 --- .github/workflows/test_cuda.yml | 2 +- .github/workflows/test_python.yml | 2 +- doc/install/easy-install.md | 8 ++++---- doc/install/install-from-source.md | 8 ++++---- 4 files changed, 10 insertions(+), 10 deletions(-) diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml index 253695f7f1..3697952988 100644 --- a/.github/workflows/test_cuda.yml +++ b/.github/workflows/test_cuda.yml @@ -47,7 +47,7 @@ jobs: - 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)') - pip install --find-links "https://www.paddlepaddle.org.cn/packages/nightly/cu126/paddlepaddle-gpu/" --index-url https://pypi.org/simple "paddlepaddle-gpu==3.3.0.dev20251204" + pip install --find-links "https://www.paddlepaddle.org.cn/packages/nightly/cu126/paddlepaddle-gpu/" --index-url https://pypi.org/simple "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' diff --git a/.github/workflows/test_python.yml b/.github/workflows/test_python.yml index fc2a929ecc..bee73c71e4 100644 --- a/.github/workflows/test_python.yml +++ b/.github/workflows/test_python.yml @@ -32,7 +32,7 @@ jobs: export TENSORFLOW_ROOT=$(python -c 'import importlib.util,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)') export PYTORCH_ROOT=$(python -c 'import torch;print(torch.__path__[0])') source/install/uv_with_retry.sh pip install --system -e .[test,jax] mpi4py --group pin_jax - source/install/uv_with_retry.sh pip install --system --find-links "https://www.paddlepaddle.org.cn/packages/nightly/cpu/paddlepaddle/" --index-url https://pypi.org/simple paddlepaddle==3.3.0.dev20251204 + source/install/uv_with_retry.sh pip install --system --find-links "https://www.paddlepaddle.org.cn/packages/nightly/cpu/paddlepaddle/" --index-url https://pypi.org/simple paddlepaddle==3.4.0.dev20260310 env: # Please note that uv has some issues with finding # existing TensorFlow package. Currently, it uses diff --git a/doc/install/easy-install.md b/doc/install/easy-install.md index 54c2068f70..6488d0f31e 100644 --- a/doc/install/easy-install.md +++ b/doc/install/easy-install.md @@ -169,9 +169,9 @@ Switch to the TensorFlow {{ tensorflow_icon }} tab for more information. ```bash # release version -pip install paddlepaddle-gpu==3.1.1 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/ +pip install paddlepaddle-gpu==3.3.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/ # nightly-build version -# pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu126/ +# pip install --pre -U paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu126/ pip install deepmd-kit ``` ::: @@ -180,9 +180,9 @@ pip install deepmd-kit ```bash # release version -pip install paddlepaddle==3.1.1 -i https://www.paddlepaddle.org.cn/packages/stable/cpu/ +pip install paddlepaddle==3.3.0 -i https://www.paddlepaddle.org.cn/packages/stable/cpu/ # nightly-build version -# pip install --pre paddlepaddle -i https://www.paddlepaddle.org.cn/packages/nightly/cpu/ +# pip install --pre -U paddlepaddle -i https://www.paddlepaddle.org.cn/packages/nightly/cpu/ pip install deepmd-kit ``` ::: diff --git a/doc/install/install-from-source.md b/doc/install/install-from-source.md index 56687c245f..ac7dc5a2c0 100644 --- a/doc/install/install-from-source.md +++ b/doc/install/install-from-source.md @@ -97,15 +97,15 @@ To install Paddle, run ```sh # cu126 # release version -pip install paddlepaddle-gpu==3.1.1 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/ +pip install paddlepaddle-gpu==3.3.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/ # nightly-build version -# pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu126/ +# pip install --pre -U paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu126/ # cpu # release version -pip install paddlepaddle==3.1.1 -i https://www.paddlepaddle.org.cn/packages/stable/cpu/ +pip install paddlepaddle==3.3.0 -i https://www.paddlepaddle.org.cn/packages/stable/cpu/ # nightly-build version -# pip install --pre paddlepaddle -i https://www.paddlepaddle.org.cn/packages/nightly/cpu/ +# pip install --pre -U paddlepaddle -i https://www.paddlepaddle.org.cn/packages/nightly/cpu/ ``` ::: From 6d05344c993f049e1813d634a312aafbd49934d1 Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Wed, 11 Mar 2026 20:31:37 +0800 Subject: [PATCH 2/2] fix adamw support --- deepmd/pd/train/training.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/deepmd/pd/train/training.py b/deepmd/pd/train/training.py index 72186cc8ed..0f1ac63663 100644 --- a/deepmd/pd/train/training.py +++ b/deepmd/pd/train/training.py @@ -165,7 +165,7 @@ def get_opt_param(params: dict[str, Any]) -> tuple[str, dict[str, Any]]: before this function is called. """ opt_type = params.get("type", "Adam") - if opt_type != "Adam": + if opt_type not in ["Adam", "AdamW"]: raise ValueError(f"Not supported optimizer type '{opt_type}'") opt_param = dict(params) opt_param.pop("type", None) @@ -647,7 +647,7 @@ def single_model_finetune( # author: iProzd # TODO add optimizers for multitask # author: iProzd - if self.opt_type == "Adam": + if self.opt_type in ["Adam", "AdamW"]: self.scheduler = paddle.optimizer.lr.LambdaDecay( learning_rate=self.lr_schedule.start_lr, lr_lambda=lambda step: ( @@ -800,7 +800,7 @@ def step(_step_id: int, task_key: str = "Default") -> None: print_str = f"Step {_step_id}: sample system{log_dict['sid']} frame{log_dict['fid']}\n" fout1.write(print_str) fout1.flush() - if self.opt_type == "Adam": + if self.opt_type in ["Adam", "AdamW"]: cur_lr = self.scheduler.get_lr() pref_lr = cur_lr