updage model container start gpu driver

nm/task_msg.go

@@ -133,7 +133,7 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
 			}
 			//taskCmd.ApiUrl = fmt.Sprintf(taskCmd.ApiUrl, container.Ports[0].PublicPort)
 			log.Info("Container ports:", container.Ports)
-			taskCmd.ApiUrl = fmt.Sprintf("http://%s:%d/aigic", ip, 5001)
+			taskCmd.ApiUrl = fmt.Sprintf("http://%s:%d/aigic", ip, container.Ports[0].PrivatePort)
 			//taskCmd.ApiUrl = fmt.Sprintf("http://%s:%d/aigic", ip, container.Ports[0].PublicPort)
 			log.Info("Setting container api url:", taskCmd.ApiUrl)
 			isImageRunExist = true