Error with PyTorch exporter: Decorator script does not work on dynamically compiled function Rank

[yuanshengpan]

ONNX: starting export with onnx 1.18.0 opset 18...
ONNX: export failure ❌ 2.2s: Decorator script does not work on dynamically compiled function Rank.
Traceback (most recent call last):
File "onnxscript/_internal/ast_utils.py", line 15, in get_src_and_ast
File "inspect.py", line 1258, in getsource
File "inspect.py", line 1240, in getsourcelines
File "inspect.py", line 1077, in findsource
OSError: could not get source code

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "test1.py", line 4, in
File "ultralytics/engine/model.py", line 728, in export
File "ultralytics/engine/exporter.py", line 457, in call
File "ultralytics/engine/exporter.py", line 196, in outer_func
File "ultralytics/engine/exporter.py", line 191, in outer_func
File "ultralytics/engine/exporter.py", line 584, in export_onnx
File "ultralytics/utils/export.py", line 35, in export_onnx
File "torch/onnx/init.py", line 296, in export
File "torch/onnx/_internal/exporter/_compat.py", line 141, in export_compat
File "torch/onnx/_internal/exporter/_registration.py", line 172, in from_torchlib
File "onnxscript/_framework_apis/torch_2_5.py", line 113, in get_torchlib_ops
File "", line 1176, in _find_and_load
File "", line 1147, in _find_and_load_unlocked
File "", line 690, in _load_unlocked
File "pyimod02_importers.py", line 457, in exec_module
File "onnxscript/function_libs/torch_lib/ops/init.py", line 16, in
File "", line 1176, in _find_and_load
File "", line 1147, in _find_and_load_unlocked
File "", line 690, in _load_unlocked
File "pyimod02_importers.py", line 457, in exec_module
File "onnxscript/function_libs/torch_lib/ops/core.py", line 35, in
File "", line 1176, in _find_and_load
File "", line 1147, in _find_and_load_unlocked
File "", line 690, in _load_unlocked
File "pyimod02_importers.py", line 457, in exec_module
File "onnxscript/function_libs/torch_lib/ops/common.py", line 29, in
File "onnxscript/main.py", line 85, in transform
File "onnxscript/_internal/ast_utils.py", line 17, in get_src_and_ast
RuntimeError: Decorator script does not work on dynamically compiled function Rank.

[justinchuby]

It seems that the module was imported twice for some reason. Could you run

wget https://raw.githubusercontent.com/pytorch/pytorch/main/torch/utils/collect_env.py
# For security purposes, please check the contents of collect_env.py before running it.
python collect_env.py

to share your setup?