From 9aed73691159365ced3e5f18f1173bc9001a6f8c Mon Sep 17 00:00:00 2001 From: Mert <101130780+mertalev@users.noreply.github.com> Date: Fri, 17 May 2024 20:00:20 -0400 Subject: [PATCH] fix(ml): openvino not working with kernel 6.7.5 or later (#9541) * add envs * move to Dockerfile --- machine-learning/Dockerfile | 3 +++ 1 file changed, 3 insertions(+) diff --git a/machine-learning/Dockerfile b/machine-learning/Dockerfile index cd1f2b2d99..afd793b033 100644 --- a/machine-learning/Dockerfile +++ b/machine-learning/Dockerfile @@ -40,6 +40,9 @@ FROM python:3.11-slim-bookworm@sha256:fc39d2e68b554c3f0a5cb8a776280c0b3d73b4c04b FROM openvino/ubuntu22_runtime:2023.3.0@sha256:176646df619032ea6c10faf842867119c393e7497b7f88b5e307e932a0fd5aa8 as prod-openvino USER root +# TODO: remove this once the image has the fix for https://github.com/intel/compute-runtime/issues/710 +ENV NEOReadDebugKeys=1 \ + OverrideGpuAddressSpace=48 FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04@sha256:2d913b09e6be8387e1a10976933642c73c840c0b735f0bf3c28d97fc9bc422e0 as prod-cuda