move pip install

Signed-off-by: Sun, Xuehao <[email protected]>
intel · Dec 11, 2024 · afdef17 · afdef17
1 parent db4a175
commit afdef17
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -25,6 +25,7 @@
 
 def is_cuda_available():
     try:
+        os.system("pip install torch")
         import torch
 
         return torch.cuda.is_available()
@@ -112,10 +113,6 @@ def detect_local_sm_architectures():
 
 
 def detect_hardware():
-    try:
-        os.system("pip install torch")
-    except:
-        print("Failed to install torch")
     if is_hpu_available():
         return "requirements-hpu.txt"
     elif is_cuda_available():