Skip to content

Commit 5084468

Browse files
committed
Removing psutil
1 parent d583f88 commit 5084468

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,6 @@ def dockerfile_for_linux(output_file):
186186
numpy \\
187187
packaging \\
188188
patchelf==0.17.2 \\
189-
psutil \\
190189
wheel>=0.35.1
191190
"""
192191

@@ -346,15 +345,14 @@ def dockerfile_for_linux(output_file):
346345

347346
df += """
348347
WORKDIR /workspace/onnxruntime
349-
ARG COMMON_BUILD_ARGS="--config ${{ONNXRUNTIME_BUILD_CONFIG}} --skip_submodule_sync --build_shared_lib \
348+
ARG COMMON_BUILD_ARGS="--config ${{ONNXRUNTIME_BUILD_CONFIG}} --parallel --skip_submodule_sync --build_shared_lib \
350349
--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"
351350
""".format(
352351
cuda_archs
353352
)
354353

355354
df += """
356-
RUN N_PARALLEL_JOBS=$( [[ $(nproc --all) < 64 ]] && echo $(nproc --all) || echo 64 ) && \\
357-
./build.sh ${{COMMON_BUILD_ARGS}} --parallel ${{N_PARALLEL_JOBS}} --update --build {}
355+
RUN ./build.sh ${{COMMON_BUILD_ARGS}} --update --build {}
358356
""".format(
359357
ep_flags
360358
)

0 commit comments

Comments
 (0)