Skip to content

Commit 293366c

Browse files
committed
Use lower
1 parent 1c29185 commit 293366c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -573,7 +573,7 @@ def preprocess_gpu_flags():
573573

574574
if FLAGS.tensorrt_home is None:
575575
if target_platform() == "rhel":
576-
if platform.machine() == "aarch64":
576+
if platform.machine().lower() == "aarch64":
577577
FLAGS.tensorrt_home = "/usr/local/cuda/targets/sbsa-linux/"
578578
else:
579579
FLAGS.tensorrt_home = "/usr/local/cuda/targets/x86_64-linux/"

0 commit comments

Comments
 (0)