Support if pbs scripts add ngpus info (#215)

Author: ShaneLogic

dpdispatcher/pbs.py

@@ -18,9 +18,11 @@ def gen_script(self, job):
     def gen_script_header(self, job):
         resources = job.resources
         pbs_script_header_dict= {}
-        pbs_script_header_dict['select_node_line']="#PBS -l select={number_node}:ncpus={cpu_per_node}:ngpus={gpu_per_node}".format(
-            number_node=resources.number_node, cpu_per_node=resources.cpu_per_node, gpu_per_node=resources.gpu_per_node
+        pbs_script_header_dict['select_node_line']="#PBS -l select={number_node}:ncpus={cpu_per_node}".format(
+            number_node=resources.number_node, cpu_per_node=resources.cpu_per_node
         )
+        if (resources.gpu_per_node != 0):
+            pbs_script_header_dict['select_node_line'] += ":ngpus={gpu_per_node}".format(gpu_per_node=resources.gpu_per_node)
         pbs_script_header_dict['queue_name_line']="#PBS -q {queue_name}".format(queue_name=resources.queue_name)
         pbs_script_header = pbs_script_header_template.format(**pbs_script_header_dict)
         return pbs_script_header