From c729f18b0d80ee06857db83a3a48a41f7dee44c1 Mon Sep 17 00:00:00 2001 From: Boris Sekachev Date: Fri, 13 Dec 2024 12:28:27 +0200 Subject: [PATCH] Fixed installation of YOLOv7 on GPU (#8824) --- .../20241212_212647_sekachev.bs_fixed_serverless_func.md | 4 ++++ serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml | 4 ++-- 2 files changed, 6 insertions(+), 2 deletions(-) create mode 100644 changelog.d/20241212_212647_sekachev.bs_fixed_serverless_func.md diff --git a/changelog.d/20241212_212647_sekachev.bs_fixed_serverless_func.md b/changelog.d/20241212_212647_sekachev.bs_fixed_serverless_func.md new file mode 100644 index 000000000000..68edcaa306c8 --- /dev/null +++ b/changelog.d/20241212_212647_sekachev.bs_fixed_serverless_func.md @@ -0,0 +1,4 @@ +### Fixed + +- Installation of YOLOv7 on GPU + () diff --git a/serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml b/serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml index d9235c458155..a1f3ac70e88d 100644 --- 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