updage model container start gpu driver

b2313b9b · duanjinfei · 1c0784bc · b2313b9b
Commit b2313b9b authored Feb 20, 2024 by duanjinfei
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

task_msg.go nm/task_msg.go +1 -0

No files found.
--- a/nm/task_msg.go
+++ b/nm/task_msg.go
@@ -152,6 +152,7 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
 			}
 			break
 		}
+		taskCmd.DockerCmd.HostIp = "0.0.0.0"
 		taskCmd.DockerCmd.HostPort = strconv.FormatInt(externalPort, 10)
 		taskCmd.ApiUrl = fmt.Sprintf(taskCmd.ApiUrl, externalPort)
 		//if int64(len(containers)) == conf.GetConfig().ContainerNum {