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"