|
|
@@ -1,11 +1,12 @@ |
|
|
|
package cloudbrain |
|
|
|
|
|
|
|
import ( |
|
|
|
"code.gitea.io/gitea/modules/timeutil" |
|
|
|
"encoding/json" |
|
|
|
"errors" |
|
|
|
"strconv" |
|
|
|
|
|
|
|
"code.gitea.io/gitea/modules/timeutil" |
|
|
|
|
|
|
|
"code.gitea.io/gitea/modules/storage" |
|
|
|
|
|
|
|
"code.gitea.io/gitea/models" |
|
|
@@ -16,7 +17,8 @@ import ( |
|
|
|
) |
|
|
|
|
|
|
|
const ( |
|
|
|
Command = `pip3 install jupyterlab==2.2.5 -i https://pypi.tuna.tsinghua.edu.cn/simple;service ssh stop;jupyter lab --no-browser --ip=0.0.0.0 --allow-root --notebook-dir="/code" --port=80 --LabApp.token="" --LabApp.allow_origin="self https://cloudbrain.pcl.ac.cn"` |
|
|
|
//Command = `pip3 install jupyterlab==2.2.5 -i https://pypi.tuna.tsinghua.edu.cn/simple;service ssh stop;jupyter lab --no-browser --ip=0.0.0.0 --allow-root --notebook-dir="/code" --port=80 --LabApp.token="" --LabApp.allow_origin="self https://cloudbrain.pcl.ac.cn"` |
|
|
|
Command = `service ssh stop;jupyter lab --no-browser --ip=0.0.0.0 --allow-root --notebook-dir="/code" --port=80 --LabApp.token="" --LabApp.allow_origin="self https://cloudbrain.pcl.ac.cn"` |
|
|
|
//CommandBenchmark = `echo "start benchmark";python /code/test.py;echo "end benchmark"` |
|
|
|
CommandBenchmark = `echo "start benchmark";cd /benchmark && bash run_bk.sh;echo "end benchmark"` |
|
|
|
CodeMountPath = "/code" |
|
|
@@ -184,7 +186,6 @@ func AdminOrImageCreaterRight(ctx *context.Context) { |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
func GenerateTask(ctx *context.Context, displayJobName, jobName, image, command, uuid, codePath, modelPath, benchmarkPath, snn4imagenetPath, brainScorePath, jobType, gpuQueue, description, branchName, bootFile, params string, benchmarkTypeID, benchmarkChildTypeID, resourceSpecId int) error { |
|
|
|
|
|
|
|
dataActualPath := setting.Attachment.Minio.RealPath + |
|
|
|