feat: improved rocm installer for navi 3x and rocm 5.5+ (and experimental navi 2x support)

2026-09-20 01:31:13 +02:00 · 2023-08-13 18:57:28 +08:00
parent 357b5dac60
commit 414acda2e9
2 changed files with 42 additions and 7 deletions
@@ -17,11 +17,6 @@ local_url = None

 errors.log.debug('Loading Torch')
 import torch # pylint: disable=C0411
-try:
-    rnd = torch.sum(torch.randn(2, 2)).to(0)
-    errors.log.debug(f'Torch init: {rnd / rnd == 1.0}') # fix silly pytorch_lightning issue
-except Exception:
-    pass
 try:
    import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import
 except Exception: