diff --git a/orttraining/orttraining/python/training/ortmodule/options.py b/orttraining/orttraining/python/training/ortmodule/options.py index df3b078788d16..b9b962acd189b 100644 --- a/orttraining/orttraining/python/training/ortmodule/options.py +++ b/orttraining/orttraining/python/training/ortmodule/options.py @@ -379,6 +379,7 @@ def _override_from_env_vars(self): import triton # noqa: F401 except ImportError: pass + self._logger.warning("triton is not found. Please install triton with `pip install triton`") else: self.enable_triton = True