Skip to content

Commit 272273a

Browse files
committed
fixup
1 parent bcd37a6 commit 272273a

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -185,11 +185,14 @@ def dockerfile_for_linux(output_file):
185185

186186
# Openvino changed the filename of the toolkit in 2025.0.0 so we need to detect this for
187187
# the release we want to install
188-
openvino_toolkit_filename = "UNKNOWN"
188+
openvino_folder_name = "UNKNOWN_FOLDER_NAME"
189+
openvino_toolkit_filename = "UNKNOWN_FILENAME"
189190
if OPENVINO_VERSION_MAP[FLAGS.ort_openvino][0].split(".")[0] >= "2025":
190-
openvino_toolkit_filename = "openvino_toolkit_ubuntu24_${OPENVINO_VERSION_WITH_BUILD_NUMBER}_x86_64.tgz"
191+
openvino_folder_name = "openvino_toolkit_ubuntu24_${OPENVINO_VERSION_WITH_BUILD_NUMBER}_x86_64"
192+
openvino_toolkit_filename = openvino_folder_name + ".tgz"
191193
else:
192-
openvino_toolkit_filename = "l_openvino_toolkit_ubuntu24_${OPENVINO_VERSION_WITH_BUILD_NUMBER}_x86_64.tgz"
194+
openvino_folder_name = "l_openvino_toolkit_ubuntu24_${OPENVINO_VERSION_WITH_BUILD_NUMBER}_x86_64"
195+
openvino_toolkit_filename = openvino_folder_name + ".tgz"
193196

194197
df += """
195198
# Step 1: Download and install core components
@@ -211,7 +214,7 @@ def dockerfile_for_linux(output_file):
211214
ENV PYTHONPATH=$INTEL_OPENVINO_DIR/python/python3.12:$INTEL_OPENVINO_DIR/python/python3:$PYTHONPATH
212215
""".format(
213216
openvino_toolkit_filename,
214-
openvino_toolkit_filename
217+
openvino_folder_name
215218
)
216219

217220
## TEMPORARY: Using the tensorrt-8.0 branch until ORT 1.9 release to enable ORT backend with TRT 8.0 support.

0 commit comments

Comments
 (0)