Fix cmake requirements for ARM64 (#8589)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
9 months ago · 2100d28638
parent a3fdcd0a35
commit 2100d28638
1 changed files with 2 additions and 2 deletions
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@ -344,8 +344,8 @@ class Exporter:
        requirements = ["onnx>=1.12.0"]
        if self.args.simplify:
            requirements += ["onnxsim>=0.4.33", "onnxruntime-gpu" if torch.cuda.is_available() else "onnxruntime"]
-        if ARM64:
-            check_requirements("cmake")  # 'cmake' is needed to build onnxsim on aarch64
+            if ARM64:
+                check_requirements("cmake")  # 'cmake' is needed to build onnxsim on aarch64
        check_requirements(requirements)
        import onnx  # noqa