diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index afe7d3e6c2..ed5cec57cb 100644 --- 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