Commit ab5ca73d authored by yihuishou's avatar yihuishou
Browse files

🚀 Add check for cuda exists

parent cbc66dda
Loading
Loading
Loading
Loading
+14 −13
Original line number Diff line number Diff line
@@ -163,14 +163,15 @@ def main():
    print(f'Version: {VERSION}')
    print(f'Branch: {BRANCH}')
    print(f"Commit hash: {commit}")
    print('Device name: ', torch.cuda.get_device_name(0))

    if hasattr(torch, 'cuda'):
        device_name = torch.cuda.get_device_name(0)
        print('Device name: ', device_name)
        print('Cuda is available: ',torch.cuda.is_available())
        print('Cuda version: ', torch.version.cuda)
    print('Cudnn is available: ', torch.backends.cudnn.is_available())
    print('Cudnn version: ', torch.backends.cudnn.version())
    print('ZLUDA is available: ', zluda_available(torch.cuda.get_device_name(0)))
        print('ZLUDA is available: ', zluda_available(device_name))

    if zluda_available(torch.cuda.get_device_name(0)):
        if zluda_available(device_name):
            torch.backends.cudnn.enabled = False
            torch.backends.cuda.enable_flash_sdp(False)
            torch.backends.cuda.enable_math_sdp(True)