Skip to content

Commit b370799

Browse files
restore to 3.0.0
1 parent db5d3b4 commit b370799

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

.github/workflows/test_cuda.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ jobs:
4747
- run: |
4848
export PYTORCH_ROOT=$(python -c 'import torch;print(torch.__path__[0])')
4949
export TENSORFLOW_ROOT=$(python -c 'import importlib.util,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)')
50-
pip install "paddlepaddle-gpu==3.1.1" -i https://www.paddlepaddle.org.cn/packages/stable/cu126/
50+
pip install "paddlepaddle-gpu==3.0.0" -i https://www.paddlepaddle.org.cn/packages/stable/cu126/
5151
source/install/uv_with_retry.sh pip install --system -v -e .[gpu,test,lmp,cu12,torch,jax] mpi4py --reinstall-package deepmd-kit
5252
env:
5353
DP_VARIANT: cuda

.github/workflows/test_python.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ jobs:
3535
source/install/uv_with_retry.sh pip install --system -e .[test,jax] mpi4py "jax==0.5.0;python_version>='3.10'"
3636
source/install/uv_with_retry.sh pip install --system -U setuptools
3737
source/install/uv_with_retry.sh pip install --system horovod --no-build-isolation
38-
source/install/uv_with_retry.sh pip install --system --pre "paddlepaddle==3.1.1" -i https://www.paddlepaddle.org.cn/packages/stable/cpu/
38+
source/install/uv_with_retry.sh pip install --system --pre "paddlepaddle==3.0.0" -i https://www.paddlepaddle.org.cn/packages/stable/cpu/
3939
env:
4040
# Please note that uv has some issues with finding
4141
# existing TensorFlow package. Currently, it uses

source/tests/pd/model/test_deeppot.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,10 @@ def setUp(self) -> None:
4242
trainer = get_trainer(deepcopy(self.config))
4343
trainer.run()
4444

45-
with paddle.device.device_guard("cpu"):
46-
input_dict, label_dict, _ = trainer.get_data(is_train=False)
45+
ori_place = paddle.device.get_device()
46+
paddle.device.set_device("cpu")
47+
input_dict, label_dict, _ = trainer.get_data(is_train=False)
48+
paddle.device.set_device(ori_place)
4749
trainer.wrapper(**input_dict, label=label_dict, cur_lr=1.0)
4850
self.model = "model.pd"
4951

0 commit comments

Comments
 (0)