Support ubuntu 23.04+ in cm docker

mlcommons · Feb 9, 2024 · 5720193 · 5720193
1 parent 12e0afa
commit 5720193
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cm-mlops/script/build-dockerfile/customize.py b/cm-mlops/script/build-dockerfile/customize.py
@@ -111,6 +111,10 @@ def preprocess(i):
     f.write('RUN '+ get_value(env, config, 'package-manager-get-cmd') + " " + " ".join(get_value(env, config,
         'packages')) + EOL)
 
+    if env['CM_DOCKER_OS'] == "ubuntu":
+        if int(env['CM_DOCKER_OS_VERSION'].split('.')[0]) >= 23:
+            if "--break-system-packages" not in env.get('CM_DOCKER_PIP_INSTALL_EXTRA_FLAGS', ''):
+              env['CM_DOCKER_PIP_INSTALL_EXTRA_FLAGS'] = " --break-system-packages"
     pip_extra_flags = env.get('CM_DOCKER_PIP_INSTALL_EXTRA_FLAGS', '')
 
     f.write(EOL+'# Install python packages' + EOL)