|
@@ -553,7 +553,7 @@ public class ProjectApplicationService {
|
|
|
remainderNodeMap.put(currentNodeName, cpuOrder);
|
|
|
}
|
|
|
// 只有准备启动(即 currentCount == 0)的时候才指定 cpu 编号
|
|
|
- log.debug("创建任务 " + taskId + " 的 yaml:是否使用 gpu (0是1否)" + isChoiceGpu + ",当前节点已创建 yaml 个数为:" + currentCount + ",当前节点名称为:" + currentNodeName + ",当前 cpu 编号为:" + cpuOrder);
|
|
|
+ log.debug("创建任务 " + taskId + " 的 yaml:是否使用 gpu (0是1否)" + isChoiceGpu + ",当前节点已创建 yaml 个数为:" + currentCount + ",当前节点名称为:" + currentNodeName + ",当前 cpu 编号为:" + cpuOrder + ",镜像名:" + algorithmDockerImage);
|
|
|
String yamlRedisKey = projectDomainService.createTempYaml(projectId, vehicleConfigId, modelType, algorithmDockerImage, currentNodeName, partition, offset, isChoiceGpu, cpuOrder);
|
|
|
if (currentCount == 0) {
|
|
|
yamlToRunRedisKeyList.add(yamlRedisKey);
|
|
@@ -760,7 +760,7 @@ public class ProjectApplicationService {
|
|
|
OsUtil.exec("docker push " + dockerImage);
|
|
|
// FileUtil.rm(algorithmTarLinuxTempPath);
|
|
|
// log.info("已删除算法临时文件:" + algorithmTarLinuxTempPath);
|
|
|
- algorithmExpandMapper.updateDockerImportAndDockerImageById(dockerImage, algorithmId);
|
|
|
+ algorithmExpandMapper.updateDockerImportAndDockerImageById("1", dockerImage, algorithmId);
|
|
|
algorithmExpandMapper.updateStatusByAlgorithmId(AlgorithmExpandEntity.builder().algorithmId(algorithmId).status(DictConstants.ALGORITHM_EXPAND_STATUS_TESTING).build());
|
|
|
} else {
|
|
|
log.info("算法镜像" + dockerImageWithoutVersion + "已导入。");
|