|
78 | 78 | "2024.0.0", # Standalone OpenVINO |
79 | 79 | "3.2.6", # DCGM version |
80 | 80 | "0.5.3.post1", # vLLM version |
81 | | - "3.10.13", # Python version |
82 | 81 | ) |
83 | 82 | } |
84 | 83 |
|
@@ -909,7 +908,7 @@ def install_dcgm_libraries(dcgm_version, target_machine): |
909 | 908 | ) |
910 | 909 |
|
911 | 910 |
|
912 | | -def create_dockerfile_buildbase_rhel(ddir, dockerfile_name, argmap, backends): |
| 911 | +def create_dockerfile_buildbase_rhel(ddir, dockerfile_name, argmap): |
913 | 912 | df = """ |
914 | 913 | ARG TRITON_VERSION={} |
915 | 914 | ARG TRITON_CONTAINER_VERSION={} |
@@ -951,7 +950,7 @@ def create_dockerfile_buildbase_rhel(ddir, dockerfile_name, argmap, backends): |
951 | 950 | libb64-devel \\ |
952 | 951 | gperftools-devel \\ |
953 | 952 | patchelf \\ |
954 | | - python3-devel \\ |
| 953 | + python3.12-devel \\ |
955 | 954 | python3-pip \\ |
956 | 955 | python3-setuptools \\ |
957 | 956 | rapidjson-devel \\ |
@@ -989,19 +988,6 @@ def create_dockerfile_buildbase_rhel(ddir, dockerfile_name, argmap, backends): |
989 | 988 | # && apt-get update -q=2 \\ |
990 | 989 | # && apt-get install -y --no-install-recommends cmake=3.27.7* cmake-data=3.27.7* |
991 | 990 | """ |
992 | | - if "python" in backends: |
993 | | - df += """ |
994 | | -# python3.10 is not available through yum for RHEL8. It must instead must be installed via pyenv. |
995 | | -# Only copy out libpython3.1* so as to not overwrite the general libpython3.so lib. |
996 | | -RUN curl https://pyenv.run | bash \\ |
997 | | - && echo 'export PYENV_ROOT="$HOME/.pyenv"' >> ~/.bashrc \\ |
998 | | - && echo '[[ -d $PYENV_ROOT/bin ]] && export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bashrc \\ |
999 | | - && echo 'eval "$(pyenv init -)"' >> ~/.bashrc |
1000 | | -RUN CONFIGURE_OPTS=\"--with-openssl=/usr/lib64\" && pyenv install {} \\ |
1001 | | - && cp /root/.pyenv/versions/{}/lib/libpython3.1* /usr/lib64/""".format( |
1002 | | - TRITON_VERSION_MAP[FLAGS.version][7], TRITON_VERSION_MAP[FLAGS.version][7] |
1003 | | - ) |
1004 | | - |
1005 | 991 | if FLAGS.enable_gpu: |
1006 | 992 | df += install_dcgm_libraries(argmap["DCGM_VERSION"], target_machine()) |
1007 | 993 | df += """ |
@@ -1404,22 +1390,10 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach |
1404 | 1390 | # Add dependencies needed for python backend |
1405 | 1391 | if "python" in backends: |
1406 | 1392 | if target_platform() == "rhel": |
1407 | | - df += """ |
1408 | | -# python3.10 is not available through yum for RHEL8. It must instead must be installed via pyenv. |
1409 | | -# Only copy out libpython3.1* so as to not overwrite the general libpython3.so lib. |
1410 | | -RUN curl https://pyenv.run | bash \\ |
1411 | | - && echo 'export PYENV_ROOT="$HOME/.pyenv"' >> ~/.bashrc \\ |
1412 | | - && echo '[[ -d $PYENV_ROOT/bin ]] && export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bashrc \\ |
1413 | | - && echo 'eval "$(pyenv init -)"' >> ~/.bashrc |
1414 | | -RUN CONFIGURE_OPTS=\"--with-openssl=/usr/lib64\" && pyenv install {} \\ |
1415 | | - && cp /root/.pyenv/versions/{}/lib/libpython3.1* /usr/lib64/""".format( |
1416 | | - TRITON_VERSION_MAP[FLAGS.version][7], |
1417 | | - TRITON_VERSION_MAP[FLAGS.version][7], |
1418 | | - ) |
1419 | | - |
1420 | 1393 | df += """ |
1421 | 1394 | # python3, python3-pip and some pip installs required for the python backend |
1422 | 1395 | RUN yum install -y \\ |
| 1396 | + python3.12-devel \\ |
1423 | 1397 | libarchive-devel \\ |
1424 | 1398 | python3-pip \\ |
1425 | 1399 | openssl-devel \\ |
@@ -1649,7 +1623,7 @@ def create_build_dockerfiles( |
1649 | 1623 |
|
1650 | 1624 | if target_platform() == "rhel": |
1651 | 1625 | create_dockerfile_buildbase_rhel( |
1652 | | - FLAGS.build_dir, "Dockerfile.buildbase", dockerfileargmap, backends |
| 1626 | + FLAGS.build_dir, "Dockerfile.buildbase", dockerfileargmap |
1653 | 1627 | ) |
1654 | 1628 | else: |
1655 | 1629 | create_dockerfile_buildbase( |
|
0 commit comments