diff --git a/DP-GEN/machine.json b/DP-GEN/machine.json
index a2a321c93d7afcad374005910a17396d81a73347..454f518ec1b7172ab2038b1a74e1c706b726908b 100644
--- a/DP-GEN/machine.json
+++ b/DP-GEN/machine.json
@@ -23,7 +23,7 @@
                 "checkpoint_files":["00*/checkpoint","00*/model.ckpt*"],
                 "checkpoint_time":30,
                 "platform": "ali",
-                "image_name":"LBG_DeePMD-kit_2.1.0_v1",
+                "image_name":"registry.dp.tech/dptech/deepmd-kit:2.1.5-cuda11.6",
                 "on_demand":0
             }
         }
@@ -56,7 +56,7 @@
               "disk_size": 200,
               "scass_type":"c4_m15_1 * NVIDIA T4",
               "platform": "ali",
-              "image_name":"LBG_DeePMD-kit_2.1.0_v1",
+              "image_name":"registry.dp.tech/dptech/deepmd-kit:2.1.5-cuda11.6",
               "checkpoint_files": "sync_files",
               "checkpoint_time":30,
               "on_demand":0