You can't run a single DeePMD-kit DPLR simulation across both a CPU-only node and a GPU node at the same time. All calculations (short-range on GPU, long-range on CPU) must happen within each MPI process, and each process expects access to both CPU and GPU resources if needed. Heterogeneous runs (splitting the workload between a CPU-only node and a GPU node) are not supported for a single simulation run—there’s no mechanism to offload the short-range part to the GPU node and the long-range part to the CPU node in one job. This is a limitation of both DeePMD-kit and LAMMPS for DPLR models source.

For best performance, run your DPLR simulation on the GPU node, using all its CPU cores and the GPU, and tune your MPI and threading variables as discussed earlier source. You can use the CPU node for other tasks (like training, post-processing, or running non-DPLR jobs) while the GPU node runs DPLR MD.

If you want to automate job submission or manage resources across both nodes, you can use tools like DPDispatcher to submit separate jobs to each node, but this does not enable a single DPLR simulation to use both nodes at once source.

Let me know if you want an optimized submission script for either node!

To reply, just mention @dosu.

Docs are dead. Just use Dosu.

Leave Feedback Ask Dosu about deepmd-kit Share Dosu with your team

bulk water

variable BETA equal 0.3
variable KMESH equal 60 #should be twice the box size in A

units metal
boundary p p p
atom_style full

neighbor 2.0 bin
neigh_modify every 10 delay 0 check no exclude type 1 5

read_data water.data

group real_atom type 1 2
group virtual_atom type 5

pair_style deepmd ./dplrcompress.pb
pair_coeff * *
bond_style zero
bond_coeff *
special_bonds lj/coul 1 1 1 angle no

kspace_style pppm/dplr 1e-5
kspace_modify gewald ${BETA} diff ik mesh ${KMESH} ${KMESH} ${KMESH}

fix 0 all dplr model ./dplrcompress.pb type_associate 1 5 bond_type 1
fix_modify 0 virial yes

compute real_temp real_atom temp
compute real_press all pressure real_temp
velocity all create 300 18234589
fix 1 all npt temp 300 300 0.05 iso 1.01325 1.01325 0.5
fix_modify 1 temp real_temp press real_press

dump 1 all custom 1000 water.dump id type x y z
dump_modify 1 append yes sort id

timestep 0.0005
thermo_style custom step vol temp pe ke etotal press
thermo_modify temp real_temp press real_press
thermo 1000
thermo_modify flush yes

run 1000
write_data out.lmp nocoeff
#############################This is my inputs###########################
My current system is an NVT ensemble bulk water. I can delete the lines "fix_modify 0 virial yes" and "compute real_press all pressure real_temp". Then, can I modify the command "fix_modify 1 temp real_temp press real_press" to "fix_modify 1 temp real_temp"?@dosu

Yes, for an NVT ensemble, you can safely delete the lines fix_modify 0 virial yes and compute real_press all pressure real_temp. You should also change fix_modify 1 temp real_temp press real_press to fix_modify 1 temp real_temp. In fact, you should also change fix 1 all npt ... to fix 1 all nvt temp 300 300 0.05, since NVT does not require pressure control—only temperature control is needed. This matches the recommended setup in the DPLR documentation and example scripts for NVT simulations, where only the temperature compute is specified for the thermostat, and pressure-related computes and modifications are omitted see example.

If you reference real_press elsewhere (like in thermo_modify), remove or adjust those as well. Your NVT input should look like:

fix 1 all nvt temp 300 300 0.05
fix_modify 1 temp real_temp

This setup ensures the thermostat acts only on the real atoms, as required for DPLR models.

To reply, just mention @dosu.

Share context across your team and agents. Try Dosu.

Leave Feedback Learn about deepmd-kit with Dosu Add Dosu to your team

Could i write tiis like "fix 1 real_atom nvt temp 300 300 0.05" ? So I can delete "fix_modify 1 temp real_temp"@dosu

You should not delete fix_modify 1 temp real_temp even if you use fix 1 real_atom nvt temp 300 300 0.05. The recommended and consistently used approach for DPLR/DeePMD-kit is to apply the thermostat to the real_atom group and also specify the temperature compute with fix_modify 1 temp real_temp. This ensures the thermostat uses the temperature of only the real atoms, excluding virtual atoms (Wannier centroids). Omitting fix_modify may cause warnings or incorrect thermostatting, and all authoritative examples and discussions include it—even when using fix nvt on a group see documentation see discussion.

So, keep both:

fix 1 real_atom nvt temp 300 300 0.05
fix_modify 1 temp real_temp

This setup is required for correct and reproducible DPLR simulations.

To reply, just mention @dosu.

Docs are dead. Just use Dosu.

Leave Feedback Ask Dosu about deepmd-kit Share Dosu with your team