Alex Cheema 10 月之前
父节点
当前提交
e93a319ca4
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      exo/topology/device_capabilities.py

+ 1 - 1
exo/topology/device_capabilities.py

@@ -43,7 +43,7 @@ def linux_device_capabilities() -> DeviceCapabilities:
     from tinygrad import Device
     
     print(f"tinygrad {Device.DEFAULT=}")
-    if Device.DEFAULT == "CUDA" or Device.DEFAULT == "NV" or Device.DEFAULT="GPU":
+    if Device.DEFAULT == "CUDA" or Device.DEFAULT == "NV" or Device.DEFAULT=="GPU":
         import pynvml, pynvml_utils
         pynvml.nvmlInit()
         handle = pynvml.nvmlDeviceGetHandleByIndex(0)