[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
ed415f5b83
commit
267601e52a
1 changed files with 5 additions and 2 deletions
|
|
@ -1180,11 +1180,14 @@ def _has_usable_nvidia_gpu() -> bool:
|
|||
for _candidate in (
|
||||
os.path.join(
|
||||
os.environ.get("ProgramFiles", r"C:\Program Files"),
|
||||
"NVIDIA Corporation", "NVSMI", "nvidia-smi.exe",
|
||||
"NVIDIA Corporation",
|
||||
"NVSMI",
|
||||
"nvidia-smi.exe",
|
||||
),
|
||||
os.path.join(
|
||||
os.environ.get("SystemRoot", r"C:\Windows"),
|
||||
"System32", "nvidia-smi.exe",
|
||||
"System32",
|
||||
"nvidia-smi.exe",
|
||||
),
|
||||
):
|
||||
if os.path.isfile(_candidate):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue