From ef2f732fd4731f961308ba227fe115d6abfd4db2 Mon Sep 17 00:00:00 2001 From: liyb Date: Mon, 26 Dec 2022 15:59:18 +0800 Subject: [PATCH] update image --- DP-GEN/machine.json | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/DP-GEN/machine.json b/DP-GEN/machine.json index a2a321c..454f518 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 -- Gitee