This commit is contained in:
Niko00 2026-03-28 00:40:12 +00:00 committed by GitHub
commit 031cb1a1b2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -184,6 +184,8 @@ def is_wsl():
return True return True
elif version.endswith("microsoft-standard-WSL2"): elif version.endswith("microsoft-standard-WSL2"):
return True return True
if os.environ.get("WSL_DISTRO_NAME"):
return True
return False return False
def get_torch_device(): def get_torch_device():