Loading launch.py +14 −13 Original line number Diff line number Diff line Loading @@ -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) Loading Loading
launch.py +14 −13 Original line number Diff line number Diff line Loading @@ -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) Loading