Fix CPU-only import

py/torch_tensorrt/_Device.py

@@ -145,7 +145,7 @@ def _from_torch_device(cls, torch_dev: torch.device) -> Device:
 
     @classmethod
     def _current_device(cls) -> Device:
-        dev_id = torch.cuda.current_device()
+        dev_id = torch.cuda.current_device() if torch.cuda.is_available() else 0
         return cls(gpu_id=dev_id)
 
     @staticmethod

py/torch_tensorrt/_utils.py

@@ -87,18 +87,24 @@ def is_tensorrt_version_supported(min_version: str) -> bool:
 
 
 def is_tegra_platform() -> bool:
+    if not torch.cuda.is_available():
+        return False
     if torch.cuda.get_device_capability() in [(8, 7), (7, 2)]:
         return True
     return False
 
 
 def is_orin() -> bool:
+    if not torch.cuda.is_available():
+        return False
     if torch.cuda.get_device_capability() in [(8, 7)]:
         return True
     return False
 
 
 def is_thor() -> bool:
+    if not torch.cuda.is_available():
+        return False
     if torch.cuda.get_device_capability() in [(11, 0)]:
         return True
     return False

py/torch_tensorrt/dynamo/_defaults.py

@@ -97,4 +97,6 @@
 
 
 def default_device() -> Device:
+    if not torch.cuda.is_available():
+        return Device(gpu_id=0)
     return Device(gpu_id=torch.cuda.current_device())