From 495270a82d1da10445a87424a2108a6d8c8da6df Mon Sep 17 00:00:00 2001
From: denghuilu <denghuilu@pku.edu.cn>
Date: Wed, 12 May 2021 21:26:35 +0800
Subject: [PATCH] add GPU UUID support for DP

---
 source/train/Local.py         | 2 +-
 source/train/RunOptions.py.in | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/source/train/Local.py b/source/train/Local.py
index d2564ff17c..1f228fe8ba 100644
--- a/source/train/Local.py
+++ b/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
diff --git a/source/train/RunOptions.py.in b/source/train/RunOptions.py.in
index 63d3544ca6..607519b87e 100644
--- a/source/train/RunOptions.py.in
+++ b/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"