Add check for WSL_DISTRO_NAME for is_wsl()

WSL supports custom kernels. Add check for WSL_DISTRO_NAME to check custom kernels.
This commit is contained in:
Niko00 2026-03-01 13:56:19 +08:00 committed by GitHub
parent 850e8b42ff
commit 6ff2489510
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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