File tree Expand file tree Collapse file tree 1 file changed +8
-8
lines changed Expand file tree Collapse file tree 1 file changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -559,13 +559,13 @@ def preprocess_gpu_flags():
559559 print ("error: linux build requires --cudnn-home and --cuda-home" )
560560
561561 if FLAGS .tensorrt_home is None :
562- if target_platform () == "rhel" :
563- if platform .machine ().lower () == "aarch64" :
564- FLAGS .tensorrt_home = "/usr/local/cuda/targets/sbsa-linux/"
565- else :
566- FLAGS .tensorrt_home = "/usr/local/cuda/targets/x86_64-linux/"
567- else :
568- FLAGS .tensorrt_home = "/usr/src/tensorrt"
562+ if target_platform () == "rhel" :
563+ if platform .machine ().lower () == "aarch64" :
564+ FLAGS .tensorrt_home = "/usr/local/cuda/targets/sbsa-linux/"
565+ else :
566+ FLAGS .tensorrt_home = "/usr/local/cuda/targets/x86_64-linux/"
567+ else :
568+ FLAGS .tensorrt_home = "/usr/src/tensorrt"
569569
570570
571571if __name__ == "__main__" :
@@ -654,4 +654,4 @@ def preprocess_gpu_flags():
654654 FLAGS .ort_openvino = None
655655 dockerfile_for_windows (FLAGS .output )
656656 else :
657- dockerfile_for_linux (FLAGS .output )
657+ dockerfile_for_linux (FLAGS .output )
You can’t perform that action at this time.
0 commit comments