실행 오류 수정
This commit is contained in:
@@ -116,8 +116,11 @@ public class DockerTrainService {
|
||||
}
|
||||
|
||||
/**
|
||||
* docker run 명령어 리스트 구성 - 환경변수 설정 - GPU 옵션 설정 - 볼륨 마운트 - 컨테이너 내부 python 실행 명령 구성 - 요청값이
|
||||
* null/blank면 해당 옵션은 "아예 생략"
|
||||
* 학습 docker run command
|
||||
*
|
||||
* @param containerName
|
||||
* @param req
|
||||
* @return
|
||||
*/
|
||||
private List<String> buildDockerRunCommand(String containerName, TrainRunRequest req) {
|
||||
|
||||
@@ -259,6 +262,11 @@ public class DockerTrainService {
|
||||
|
||||
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);
|
||||
pb.redirectErrorStream(true);
|
||||
|
||||
@@ -310,6 +318,13 @@ public class DockerTrainService {
|
||||
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) {
|
||||
|
||||
String uuid = req.getUuid();
|
||||
@@ -324,7 +339,7 @@ public class DockerTrainService {
|
||||
c.add("docker");
|
||||
c.add("run");
|
||||
c.add("--name");
|
||||
c.add(containerName);
|
||||
c.add(containerName + "=" + req.getUuid().substring(0, 8));
|
||||
c.add("--rm");
|
||||
|
||||
c.add("--gpus");
|
||||
|
||||
Reference in New Issue
Block a user