diff --git a/lisa/sut_orchestrator/azure/features.py b/lisa/sut_orchestrator/azure/features.py
index 9142f1e937..4d38b1924d 100644
--- a/lisa/sut_orchestrator/azure/features.py
+++ b/lisa/sut_orchestrator/azure/features.py
@@ -28,7 +28,7 @@
 from lisa.features.gpu import ComputeSDK
 from lisa.features.resize import ResizeAction
 from lisa.node import Node, RemoteNode
-from lisa.operating_system import CentOs, Redhat, Suse, Ubuntu
+from lisa.operating_system import Redhat, Suse, Ubuntu
 from lisa.search_space import RequirementMethod
 from lisa.tools import Dmesg, Lspci, Modprobe
 from lisa.util import (
@@ -170,7 +170,7 @@ class Gpu(AzureFeatureMixin, features.Gpu):
         {
             "publisher": "Microsoft.HpcCompute",
             "type": "NvidiaGpuDriverLinux",
-            "typeHandlerVersion": "1.6",
+            "typeHandlerVersion": "1.7",
             "autoUpgradeMinorVersion": true,
             "settings": {
             }
@@ -180,11 +180,16 @@ class Gpu(AzureFeatureMixin, features.Gpu):
 
     def is_supported(self) -> bool:
         # TODO: more supportability can be defined here
-        supported_distro = (CentOs, Redhat, Ubuntu, Suse)
-        if isinstance(self._node.os, supported_distro):
-            return True
-
-        return False
+        node = self._node
+        supported = False
+        if isinstance(node.os, Redhat):
+            supported = node.os.information.version >= "7.0.0"
+        elif isinstance(node.os, Ubuntu):
+            supported = node.os.information.version >= "18.0.0"
+        elif isinstance(node.os, Suse):
+            supported = node.os.information.version >= "15.0.0"
+
+        return supported
 
     def get_supported_driver(self) -> List[ComputeSDK]:
         driver_list = []