Fix regression caused by previous commit

lassoan · lassoan · commit 12ed0b2777a4 · 2022-12-23T23:42:36.000-05:00
Fixes the problem: Command ‘[‘C:/Users/Akmal/AppData/Local/NA-MIC/Slicer 5.2.1/bin/…/bin\PythonSlicer.EXE’, ‘-m’, ‘light_the_torch’, ‘install’, ‘torch’]’ returned non-zero exit status 1.
diff --git a/PyTorchUtils/PyTorchUtils.py b/PyTorchUtils/PyTorchUtils.py
@@ -110,6 +110,7 @@ def nvidiaDriverVersionInformation(self):
       import light_the_torch._cb as computationBackend
       return f"installed version {str(computationBackend._detect_nvidia_driver_version())}"
     except Exception as e:
+      # Don't install light-the-torch just for getting the NVIDIA driver version
       return ""
 
   @property
@@ -177,6 +178,12 @@ def installTorch(self, askConfirmation=False, forceComputationBackend=None):
         logging.info('Installation of PyTorch aborted by user')
         return None
 
+    try:
+      import light_the_torch._patch
+    except:
+      PyTorchUtilsLogic._installLightTheTorch()
+      import light_the_torch._patch
+
     slicer.util._executePythonModule('light_the_torch', args)
     import torch
     logging.info(f'PyTorch {torch.__version__} installed successfully.')
