ModuleNotFoundError: No module named 'torch.fx._compatibility'

microsoft / nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

https://nni.readthedocs.io

MIT License

13.88k stars 1.81k forks source link

ModuleNotFoundError: No module named 'torch.fx._compatibility' #5762

Closed 99999-LQL closed 3 months ago

99999-LQL commented 3 months ago

Describe the issue: torch:1.9

Environment:

NNI version:
Training service (local|remote|pai|aml|etc):
Client OS:
Server OS (for remote mode only):
Python version:
PyTorch/TensorFlow version:
Is conda/virtualenv/venv used?:
Is running in Docker?:

Configuration:

Experiment config (remember to remove secrets!):
Search space:

Log message:

nnimanager.log:
dispatcher.log:
nnictl stdout and stderr:

How to reproduce it?: