diff --git a/optimum/utils/import_utils.py b/optimum/utils/import_utils.py
index 8da1df5fac..c3c0cc6aae 100644
--- a/optimum/utils/import_utils.py
+++ b/optimum/utils/import_utils.py
@@ -113,6 +113,8 @@ def _is_package_available(
         "onnxruntime-migraphx",
         "ort-migraphx-nightly",
         "ort-rocm-nightly",
+        # For DirectML
+        "onnxruntime-directml",
     ],
 )
 _tf_available, _tf_version = _is_package_available(