Merge pull request #620 from denghuilu/master

add GPU UUID support for DP

Author: amcadmus

source/train/Local.py

@@ -7,7 +7,7 @@ def get_resource ():
     if gpus is not None :
         if gpus != "" :
             gpus = gpus.split(",")
-            gpus = [int(ii) for ii in gpus]
+            gpus = [ii for ii in gpus]
         else :
             gpus = []
     return nodename, nodelist, gpus

source/train/RunOptions.py.in

@@ -227,7 +227,7 @@ class RunOptions (object) :
         self.my_task_index = 0
         self.my_socket = None
         if gpus is not None and len(gpus) > 0:
-            self.my_device = "gpu:%d" % gpus[0]
+            self.my_device = "gpu:" + str(gpus[0])
             # self.my_device = "gpu:0"
         else :
             self.my_device = "cpu:0"