Fixed installation of YOLOv7 on GPU (#8824)

cvat-ai · Dec 13, 2024 · c729f18 · c729f18
1 parent ee98635
commit c729f18
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/changelog.d/20241212_212647_sekachev.bs_fixed_serverless_func.md b/changelog.d/20241212_212647_sekachev.bs_fixed_serverless_func.md
@@ -0,0 +1,4 @@
+### Fixed
+
+- Installation of YOLOv7 on GPU
+  (<https://github.com/cvat-ai/cvat/pull/8824>)
diff --git a/serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml b/serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml
@@ -95,7 +95,7 @@ spec:
   eventTimeout: 30s
   build:
     image: cvat.onnx.wongkinyiu.yolov7:latest-gpu
-    baseImage: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
+    baseImage: nvidia/cuda:12.6.3-cudnn-runtime-ubuntu22.04
 
     directives:
       preCopy:
@@ -106,7 +106,7 @@ spec:
         - kind: WORKDIR
           value: /opt/nuclio
         - kind: RUN
-          value: pip install onnxruntime-gpu=='1.16.*' opencv-python-headless pillow pyyaml
+          value: pip install onnxruntime-gpu=='1.20.*' opencv-python-headless pillow pyyaml
         - kind: WORKDIR
           value: /opt/nuclio
         - kind: RUN