We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1c29185 commit 293366cCopy full SHA for 293366c
tools/gen_ort_dockerfile.py
@@ -573,7 +573,7 @@ def preprocess_gpu_flags():
573
574
if FLAGS.tensorrt_home is None:
575
if target_platform() == "rhel":
576
- if platform.machine() == "aarch64":
+ if platform.machine().lower() == "aarch64":
577
FLAGS.tensorrt_home = "/usr/local/cuda/targets/sbsa-linux/"
578
else:
579
FLAGS.tensorrt_home = "/usr/local/cuda/targets/x86_64-linux/"
0 commit comments