[Bug] No module named 'torch._dynamo.functional_export'

[leeeizhang]

I have installed pytorch==2.8.0+cu128, but meet following issues when running the examples:

(autoparallel) ~/xxx/autoparallel$ python examples/example_autoparallel.py
Traceback (most recent call last):
  File "/xxx/autoparallel/examples/example_autoparallel.py", line 16, in <module>
    from autoparallel.api import AutoParallel
  File "/xxx/autoparallel/autoparallel/api.py", line 14, in <module>
    from torch._dynamo.functional_export import _dynamo_graph_capture_for_export
ModuleNotFoundError: No module named 'torch._dynamo.functional_export'

I guess it might be caused by my mismatched pytorch version (2.8.0.dev20250506).

is there any link to install the old nightly pytorch (2.8.0.dev20250506) ? since this dev version could not be found in https://download.pytorch.org/whl/nightly/cu128/torch.

[fmassa]

Hi,

Thanks for the report. We need to update the documentation.

The documentation is now actually outdated. We now require a latest PyTorch nightly for AutoParallel to run.
Can you try updating to latest PyTorch and report back?

Thanks!