Skip to content

Produce ONNX wheel package as part of the build #320

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 12 commits into from
Oct 9, 2025
Merged

Produce ONNX wheel package as part of the build #320

Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
3ab90d7
Updateing commn properties
mc-nv Oct 8, 2025
4dc816a
Removing skip tests
mc-nv Oct 8, 2025
59a6b95
Adding missed packages. Sort installed packages.
mc-nv Oct 8, 2025
2d751b0
Keep existing set of codes for Win32
mc-nv Oct 8, 2025
9b6bf20
Set parallel count.
mc-nv Oct 8, 2025
5626515
Update gen script
mc-nv Oct 8, 2025
8c0fce9
Stick to CPU count
mc-nv Oct 8, 2025
77f2a73
Remove double bracers
mc-nv Oct 8, 2025
d2de8cf
Set max thread count to 64 to prevent OOM in CI
mc-nv Oct 8, 2025
d583f88
Add extra curtly bracers to adress Python syntax
mc-nv Oct 8, 2025
5084468
Removing psutil
mc-nv Oct 8, 2025
6796e68
Make CMake output verbose for image
mc-nv Oct 9, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
63 changes: 35 additions & 28 deletions tools/gen_ort_dockerfile.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -134,16 +134,16 @@ def dockerfile_for_linux(output_file):
ENV PYTHON_BIN_PATH=${PYBIN}/python${PYVER} \
PATH=${PYBIN}:${PATH}

RUN yum install -y \
wget \
zip \
ca-certificates \
curl \
python3-pip \
git \
gnupg \
RUN yum install -y \\
ca-certificates \\
curl \\
git \\
gnupg \\
gnupg1 \
openssl-devel
openssl-devel \\
python3-pip \
wget \\
zip

RUN pip3 install patchelf==0.17.2
"""
Expand All @@ -166,22 +166,29 @@ def dockerfile_for_linux(output_file):

df += """

RUN apt-get update && apt-get install -y --no-install-recommends \
software-properties-common \
wget \
zip \
ca-certificates \
build-essential \
curl \
libcurl4-openssl-dev \
libssl-dev \
python3-dev \
python3-pip \
git \
gnupg \
gnupg1

RUN pip3 install patchelf==0.17.2 cmake==4.0.3
RUN apt-get update && apt-get install -y --no-install-recommends \\
build-essential \\
ca-certificates \\
curl \\
git \\
gnupg \\
gnupg1 \\
libcurl4-openssl-dev \\
libssl-dev \\
python3-dev \\
python3-pip \\
software-properties-common \\
wget \\
zip

RUN pip3 install \\
cmake==4.0.3 \\
numpy \\
packaging \\
patchelf==0.17.2 \\
wheel>=0.35.1

ENV VERBOSE=1
"""

if FLAGS.ort_openvino is not None:
Expand Down Expand Up @@ -340,8 +347,8 @@ def dockerfile_for_linux(output_file):

df += """
WORKDIR /workspace/onnxruntime
ARG COMMON_BUILD_ARGS="--config ${{ONNXRUNTIME_BUILD_CONFIG}} --skip_submodule_sync --parallel --build_shared_lib \
--compile_no_warning_as_error --build_dir /workspace/build --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES='{}' --cmake_extra_defines CMAKE_POLICY_VERSION_MINIMUM=3.5 "
ARG COMMON_BUILD_ARGS="--config ${{ONNXRUNTIME_BUILD_CONFIG}} --parallel --skip_submodule_sync --build_shared_lib \
--compile_no_warning_as_error --build_dir /workspace/build --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES='{}' --cmake_extra_defines CMAKE_POLICY_VERSION_MINIMUM=3.5 --build_wheel"
""".format(
cuda_archs
)
Expand Down Expand Up @@ -523,7 +530,7 @@ def dockerfile_for_windows(output_file):
WORKDIR /workspace/onnxruntime
ARG VS_DEVCMD_BAT="\\BuildTools\\VC\\Auxiliary\\Build\\vcvars64.bat"
RUN powershell Set-Content 'build.bat' -value 'call %VS_DEVCMD_BAT%',(Get-Content 'build.bat')
RUN build.bat --cmake_generator "Visual Studio 17 2022" --config Release --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=75;80;86;90;100;120" --skip_submodule_sync --parallel --build_shared_lib --compile_no_warning_as_error --skip_tests --update --build --build_dir /workspace/build {}
RUN build.bat --cmake_generator "Visual Studio 17 2022" --config Release --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=75;80;86;90;100;120" --skip_submodule_sync --parallel --build_shared_lib --compile_no_warning_as_error --skip_tests --build_wheel --update --build --build_dir /workspace/build {}
""".format(
ep_flags
)
Expand Down
Loading