Python3.12

Author: fpetrini15

build.py

@@ -78,6 +78,7 @@
         "2024.0.0",  # Standalone OpenVINO
         "3.2.6",  # DCGM version
         "0.5.3.post1",  # vLLM version
+        "3.12",  # RHEL Python version
     )
 }
 
@@ -925,6 +926,7 @@ def create_dockerfile_buildbase_rhel(ddir, dockerfile_name, argmap):
 ARG TRITON_VERSION
 ARG TRITON_CONTAINER_VERSION
 """
+    df += change_default_python_version_rhel(TRITON_VERSION_MAP[FLAGS.version][7])
     df += """
 # Install docker docker buildx
 RUN yum install -y ca-certificates curl gnupg yum-utils \\
@@ -1390,14 +1392,17 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
     # Add dependencies needed for python backend
     if "python" in backends:
         if target_platform() == "rhel":
+            df += change_default_python_version_rhel(
+                TRITON_VERSION_MAP[FLAGS.version][7]
+            )
             df += """
 # python3, python3-pip and some pip installs required for the python backend
 RUN yum install -y \\
         python3.12-devel \\
         libarchive-devel \\
         python3-pip \\
         openssl-devel \\
-        readline-devel
+        readline-devel \\
       && pip3 install --upgrade pip \\
       && pip3 install --upgrade \\
             wheel \\
@@ -1531,6 +1536,23 @@ def add_cpu_libs_to_linux_dockerfile(backends, target_machine):
     return df
 
 
+def change_default_python_version_rhel(version):
+    df = """
+# RHEL image has several python versions. It's important
+# to set the correct version, otherwise, packages that are
+# pip installed will not be found during testing.
+ENV PYVER {}
+ENV PYTHONPATH /opt/python/v
+RUN ln -sf /opt/python/cp${{PYVER/./}}* ${{PYTHONPATH}}
+ENV PYBIN ${{PYTHONPATH}}/bin
+ENV PYTHON_BIN_PATH ${{PYBIN}}/python${{PYVER}}
+ENV PATH ${{PYBIN}}:${{PATH}}
+""".format(
+        version
+    )
+    return df
+
+
 def create_dockerfile_windows(
     ddir, dockerfile_name, argmap, backends, repoagents, caches
 ):