diff --git a/src/job_manager.go b/src/job_manager.go
index 15cd925..f5c53f7 100644
--- a/src/job_manager.go
+++ b/src/job_manager.go
@@ -115,7 +115,9 @@ func (jm *JobManager) start() {
 			v.Set("cpu_limit", strconv.Itoa(jm.job.Tasks[i].NumberCPU))
 			v.Set("network", network)
 			v.Set("should_wait", "1")
-			v.Set("HDFS_path", "1")
+			v.Set("output_dir", "/output/")
+			v.Set("hdfs_dir", "http://hdfs-master:50070/user/yao/output/"+jm.job.Name)
+			v.Set("gpu_mem", strconv.Itoa(jm.job.Tasks[i].MemoryGPU))
 
 			resp, err := doRequest("POST", "http://"+jm.resources[i].ClientHost+":8000/create", strings.NewReader(v.Encode()), "application/x-www-form-urlencoded", "")
 			if err != nil {