Fix building from source on WSL (#3112)

2024-03-01 03:13:58 +08:00 · 2024-03-01 03:13:58 +08:00 · 2c08ff23c0
commit 2c08ff23c0
parent bfdcfa6a05
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -36,7 +36,7 @@ def _is_neuron() -> bool:
    torch_neuronx_installed = True
    try:
        subprocess.run(["neuron-ls"], capture_output=True, check=True)
-    except FileNotFoundError:
+    except (FileNotFoundError, PermissionError):
        torch_neuronx_installed = False
    return torch_neuronx_installed