Compare commits

...
7 Commits
4 changed files with 29 additions and 8 deletions
+13 -4
View File
@@ -24,10 +24,12 @@ RUN pip wheel --no-cache-dir --no-deps --wheel-dir /usr/src/app/wheels -r requir
# pull official base image # pull official base image
FROM dvcorg/cml:0-dvc2-base1-gpu FROM dvcorg/cml:0-dvc2-base1-gpu
# create app home # create home directory and app user
ENV APP_HOME = /home/app RUN mkdir -p /home/app && \
RUN mkdir -p $APP_HOME addgroup --system app && \
WORKDIR $APP_HOME adduser --system --group app
# add folder to store repo in
RUN mkdir -p /home/app/repo RUN mkdir -p /home/app/repo
# install packages from builder # install packages from builder
@@ -36,7 +38,14 @@ COPY --from=builder /usr/src/app/requirements.txt requirements.txt
RUN pip install --no-cache /wheels/* RUN pip install --no-cache /wheels/*
# copy in files # copy in files
WORKDIR /home/app
COPY ./code . COPY ./code .
# change ownership to app user
RUN chown -R app:app /home/app
# change to the app user
USER app
# start execution # start execution
ENTRYPOINT [ "python", "main.py" ] ENTRYPOINT [ "python", "main.py" ]
+7 -2
View File
@@ -79,8 +79,13 @@ def handle_event(request: GiteaRequest):
requirements_path = local_repo_dir / pipeline.requirements requirements_path = local_repo_dir / pipeline.requirements
os.system(f'pip install -r {requirements_path}') os.system(f'pip install -r {requirements_path}')
for cmd in pipeline.script: for cmd in pipeline.script:
print(cmd) try:
os.system(cmd)
except Exception as e:
raise HTTPException(
status_code=503,
detail=f'failed running {cmd}: {e}'
)
# send response # send response
raise HTTPException( raise HTTPException(
status_code=202, status_code=202,
+7
View File
@@ -11,3 +11,10 @@ services:
- LISTEN_IP=0.0.0.0 - LISTEN_IP=0.0.0.0
- LISTEN_PORT=8000 - LISTEN_PORT=8000
- REPO_DIR=/home/app/repo/ - REPO_DIR=/home/app/repo/
# deploy:
# resources:
# reservations:
# devices:
# - driver: nvidia
# device_ids: ['0']
# capabilities: [gpu]
+1 -1
View File
@@ -4,7 +4,7 @@ definitions:
name: test name: test
requirements: req-prod.txt requirements: req-prod.txt
script: script:
- echo nvidia-smi - nvidia-smi
pipelines: pipelines:
default: default: