diff --git a/deepmd/cluster/local.py b/deepmd/cluster/local.py
index 6737c89785..37a2e99b6e 100644
--- a/deepmd/cluster/local.py
+++ b/deepmd/cluster/local.py
@@ -21,6 +21,6 @@ def get_resource() -> Tuple[str, List[str], Optional[List[int]]]:
     if not gpus_env:
         gpus = None
     else:
-        gpus = [int(gpu) for gpu in gpus_env.split(",")]
+        gpus = [gpu for gpu in gpus_env.split(",")]
 
     return nodename, nodelist, gpus
diff --git a/deepmd/train/run_options.py b/deepmd/train/run_options.py
index cdfea06ff8..25029c4308 100644
--- a/deepmd/train/run_options.py
+++ b/deepmd/train/run_options.py
@@ -346,7 +346,7 @@ def _init_serial(self):
         self.server = None
 
         if gpus is not None:
-            self.my_device = f"gpu:{gpus[0]:d}"
+            self.my_device = "gpu:" + str(gpus[0])
         else:
             self.my_device = "cpu:0"