Merge pull request '실행 오류 수정' (#43) from feat/training_260202 into develop
Reviewed-on: #43
This commit was merged in pull request #43.
This commit is contained in:
@@ -116,8 +116,11 @@ public class DockerTrainService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* docker run 명령어 리스트 구성 - 환경변수 설정 - GPU 옵션 설정 - 볼륨 마운트 - 컨테이너 내부 python 실행 명령 구성 - 요청값이
|
* 학습 docker run command
|
||||||
* null/blank면 해당 옵션은 "아예 생략"
|
*
|
||||||
|
* @param containerName
|
||||||
|
* @param req
|
||||||
|
* @return
|
||||||
*/
|
*/
|
||||||
private List<String> buildDockerRunCommand(String containerName, TrainRunRequest req) {
|
private List<String> buildDockerRunCommand(String containerName, TrainRunRequest req) {
|
||||||
|
|
||||||
@@ -259,6 +262,11 @@ public class DockerTrainService {
|
|||||||
|
|
||||||
List<String> cmd = buildDockerEvalCommand(containerName, req);
|
List<String> cmd = buildDockerEvalCommand(containerName, req);
|
||||||
|
|
||||||
|
log.info("=== Docker Test Command ===");
|
||||||
|
log.info("Container: {}", containerName);
|
||||||
|
log.info("Command: {}", String.join(" ", cmd));
|
||||||
|
log.info("================================");
|
||||||
|
|
||||||
ProcessBuilder pb = new ProcessBuilder(cmd);
|
ProcessBuilder pb = new ProcessBuilder(cmd);
|
||||||
pb.redirectErrorStream(true);
|
pb.redirectErrorStream(true);
|
||||||
|
|
||||||
@@ -310,6 +318,13 @@ public class DockerTrainService {
|
|||||||
return new TrainRunResult(null, containerName, exit, exit == 0 ? "SUCCESS" : "FAILED", logs);
|
return new TrainRunResult(null, containerName, exit, exit == 0 ? "SUCCESS" : "FAILED", logs);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 테스트 docker run command
|
||||||
|
*
|
||||||
|
* @param containerName
|
||||||
|
* @param req
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
private List<String> buildDockerEvalCommand(String containerName, EvalRunRequest req) {
|
private List<String> buildDockerEvalCommand(String containerName, EvalRunRequest req) {
|
||||||
|
|
||||||
String uuid = req.getUuid();
|
String uuid = req.getUuid();
|
||||||
@@ -324,7 +339,7 @@ public class DockerTrainService {
|
|||||||
c.add("docker");
|
c.add("docker");
|
||||||
c.add("run");
|
c.add("run");
|
||||||
c.add("--name");
|
c.add("--name");
|
||||||
c.add(containerName);
|
c.add(containerName + "=" + req.getUuid().substring(0, 8));
|
||||||
c.add("--rm");
|
c.add("--rm");
|
||||||
|
|
||||||
c.add("--gpus");
|
c.add("--gpus");
|
||||||
|
|||||||
Reference in New Issue
Block a user