xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix cmake requirements for ARM64 (#8589)

Browse Source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Lakshantha Dissanayake 2024-03-02 07:03:21 -08:00 committed by GitHub
parent a3fdcd0a35
commit 2100d28638
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ultralytics/engine/exporter.py
View File

@ -344,8 +344,8 @@ class Exporter:
requirements = ["onnx>=1.12.0"] requirements = ["onnx>=1.12.0"]
if self.args.simplify: if self.args.simplify:
requirements += ["onnxsim>=0.4.33", "onnxruntime-gpu" if torch.cuda.is_available() else "onnxruntime"] requirements += ["onnxsim>=0.4.33", "onnxruntime-gpu" if torch.cuda.is_available() else "onnxruntime"]
if ARM64: if ARM64:
check_requirements("cmake") # 'cmake' is needed to build onnxsim on aarch64 check_requirements("cmake") # 'cmake' is needed to build onnxsim on aarch64
check_requirements(requirements) check_requirements(requirements)
import onnx # noqa import onnx # noqa