Fix YOLOv5 serverless function (#4344)

4 years ago · 2f1e89cc3b
parent f719648bb0
commit 2f1e89cc3b
1 changed files with 3 additions and 1 deletions
--- a/serverless/pytorch/ultralytics/yolov5/nuclio/function.yaml
+++ b/serverless/pytorch/ultralytics/yolov5/nuclio/function.yaml
@ -96,12 +96,14 @@ spec:
  eventTimeout: 30s
  build:
    image: cvat/ultralytics-yolov5
-    baseImage: ultralytics/yolov5:latest
+    baseImage: ultralytics/yolov5:latest-cpu
    directives:
      preCopy:
        - kind: USER
          value: root
        - kind: RUN
          value: apt update && apt install --no-install-recommends -y libglib2.0-0
        - kind: WORKDIR
          value: /opt/nuclio