Add Trains-Agent docker mode enable/disable (#5)

clearml · Nov 12, 2020 · 3499a7a · 3499a7a
1 parent 42b0419
commit 3499a7a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/trains-server-chart/templates/trains-agent-deployment.yaml b/trains-server-chart/templates/trains-agent-deployment.yaml
@@ -79,7 +79,8 @@ spec:
              curl -sSL https://get.docker.com/ | sh ;
              python3 -m pip install -U pip ;
              python3 -m pip install trains-agent{{ .Values.agent.agentVersion}} ;
-             TRAINS_DOCKER_SKIP_GPUS_FLAG=1 TRAINS_AGENT_K8S_HOST_MOUNT=/root/.trains:/root/.trains trains-agent daemon --docker {{ .Values.agent.defaultBaseDocker}} --force-current-version"
+             {{if .Values.agent.dockerMode}}TRAINS_DOCKER_SKIP_GPUS_FLAG=1 TRAINS_AGENT_K8S_HOST_MOUNT=/root/.trains:/root/.trains trains-agent daemon --docker {{ .Values.agent.defaultBaseDocker}} --force-current-version"
+             {{ else}}TRAINS_AGENT_K8S_HOST_MOUNT=/root/.trains:/root/.trains trains-agent daemon"{{ end}}
       volumes:
       - name: dockersock
         hostPath:

diff --git a/trains-server-chart/values.yaml b/trains-server-chart/values.yaml
@@ -43,6 +43,7 @@ agent:
   numberOfTrainsAgents: 0
   nvidiaGpusPerAgent: 1
   defaultBaseDocker: "nvidia/cuda"
+  dockerMode: true
   agentVersion: ""
   trainsApiHost: "https://api.trains.mydomain.com"
   trainsWebHost: "https://app.trains.mydomain.com"