Skip to content

Commit fb08b80

Browse files
amcadmusHan Wang
andauthored
Incorperate Yuzhis change to 4-GPU example (#412)
Co-authored-by: Han Wang <wang_han@iapcm.ac.cn>
1 parent 5af1fd5 commit fb08b80

File tree

1 file changed

+8
-18
lines changed

1 file changed

+8
-18
lines changed

examples/machine/DeePMD-kit-1.0/machine-local-4GPU.json

Lines changed: 8 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616
"resources":{
1717
"_comment" : "The number of nodes.",
1818
"numb_node": 1,
19-
"_comment" : "The number of GPUs.",
20-
"numb_gpu": 1,
19+
"_comment" : "If you choose to run with multiple GPUs simultaneously, just ignore numb_gpu.",
20+
"numb_gpu": 0,
2121
"_comment" : "The number of CPUs.",
2222
"task_per_node": 4,
2323
"_comment" : "The number of GPUs that can be used for each task.",
@@ -26,16 +26,9 @@
2626
"manual_cuda_multiplicity":1,
2727
"_comment" : "Allow the multi-GPU task running.",
2828
"cuda_multi_task": true,
29-
"_comment" : "Partition.",
30-
"partition": "gpu",
31-
"_comment" : "Memory limit.",
32-
"mem_limit": 64,
33-
"_comment" : "None for local device",
34-
"exclude_list": [],
3529
"module_list": [],
36-
"source_list": ["/opt/intel/parallel_studio_xe_2020/psxevars.sh"],
37-
"_comment" : "Time limit.",
38-
"time_limit": "23:0:0"
30+
"_comment" : "Environment to be activated. This will generate source xxx/psxevars.sh in scripts. ",
31+
"source_list": ["/opt/intel/parallel_studio_xe_2020/psxevars.sh"]
3932
},
4033
"_comment" : "DP-GEN will put 4 tasks together in one submitting script.",
4134
"group_size": 4
@@ -50,17 +43,13 @@
5043
},
5144
"resources": {
5245
"numb_node": 1,
53-
"numb_gpu": 4,
46+
"numb_gpu": 0,
5447
"task_per_node": 4,
5548
"manual_cuda_devices": 4,
5649
"manual_cuda_multiplicity":1,
5750
"cuda_multi_task": true,
58-
"partition": "gpu",
59-
"exclude_list": [],
60-
"mem_limit": 64,
6151
"source_list": [],
62-
"module_list": [],
63-
"time_limit": "23:0:0"
52+
"module_list": []
6453
},
6554
"command": "/home/user/Soft/Deepmd/lammps-stable_29Oct2020/src/lmp_mpi",
6655
"group_size": 4
@@ -80,7 +69,8 @@
8069
"with_mpi": true,
8170
"_comment" : "Load the intel compiler.",
8271
"source_list": ["/opt/intel/parallel_studio_xe_2020/psxevars.sh"],
83-
"envs": {"PATH" : "/home/user/Soft/VASP/vasp.5.4.4-allbak/bin:$PATH"}
72+
"envs": {"PATH" : "/home/user/Soft/VASP/vasp.5.4.4-allbak/bin:$PATH"},
73+
"_comment" : "This will generate export PATH=/home/user/Soft/VASP/vasp.5.4.4-allbak/bin:$PATH in scripts;"
8474
},
8575
"command": "vasp_std",
8676
"group_size": 1

0 commit comments

Comments
 (0)