Skip to content

Commit 705f157

Browse files
mc-nvpvijayakrish
authored andcommitted
Upadate default branch post 24.12 (#290)
* Update README and versions for 2.53.0 / 24.12 (#288) * Update 'gen_ort_dockerfile.py' fil to meet changes in ONNX Runtime 1.20.x * remove psutils * Update reformating * restore description * Update OpenVINO to 2024.5 (#287) * Update 'gen_ort_dockerfile.py' fil to meet changes in ONNX Runtime 1.20.x * remove psutils * Add OpenVINO version * Fix pre-commit issues * Extract archive in different location (#289) * Extract archive in different location * Revert "Extract archive in different location" This reverts commit e57256a. * Update installation instructions * Reapply "Extract archive in different location" This reverts commit 59e24e4.
1 parent 139178c commit 705f157

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -161,16 +161,6 @@ def dockerfile_for_linux(output_file):
161161
&& apt-get install -y --no-install-recommends cmake=3.28.3* cmake-data=3.28.3* \\
162162
&& cmake --version
163163
164-
"""
165-
if FLAGS.enable_gpu:
166-
df += """
167-
# Allow configure to pick up cuDNN where it expects it.
168-
# (Note: $CUDNN_VERSION is defined by base image)
169-
RUN _CUDNN_VERSION=$(echo $CUDNN_VERSION | cut -d. -f1-2) && \
170-
mkdir -p /usr/local/cudnn-$_CUDNN_VERSION/cuda/include && \
171-
ln -s /usr/include/cudnn.h /usr/local/cudnn-$_CUDNN_VERSION/cuda/include/cudnn.h && \
172-
mkdir -p /usr/local/cudnn-$_CUDNN_VERSION/cuda/lib64 && \
173-
ln -s /etc/alternatives/libcudnn_so /usr/local/cudnn-$_CUDNN_VERSION/cuda/lib64/libcudnn.so
174164
"""
175165

176166
if FLAGS.ort_openvino is not None:
@@ -611,6 +601,7 @@ def preprocess_gpu_flags():
611601
default=None,
612602
help='Target for build, can be "linux", "windows", "rhel", or "igpu". If not specified, build targets the current platform.',
613603
)
604+
614605
parser.add_argument(
615606
"--cuda-version", type=str, required=False, help="Version for CUDA."
616607
)

0 commit comments

Comments
 (0)