From 86caf734182767fab09e3819e656b83423c45c25 Mon Sep 17 00:00:00 2001
From: Victor Skvortsov <vds003@gmail.com>
Date: Fri, 24 Oct 2025 16:29:57 +0500
Subject: [PATCH] Detect nvidia inside WSL2

---
 runner/internal/common/gpu.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/runner/internal/common/gpu.go b/runner/internal/common/gpu.go
index 19c46ea661..045cc773be 100644
--- a/runner/internal/common/gpu.go
+++ b/runner/internal/common/gpu.go
@@ -16,6 +16,8 @@ const (
 )
 
 func GetGpuVendor() GpuVendor {
+	// FIXME: There might be errors other than os.ErrNotExist that are ignored silently.
+	// Propagate and log.
 	if _, err := os.Stat("/dev/kfd"); !errors.Is(err, os.ErrNotExist) {
 		return GpuVendorAmd
 	}
@@ -28,5 +30,11 @@ func GetGpuVendor() GpuVendor {
 	if _, err := os.Stat("/dev/tenstorrent"); !errors.Is(err, os.ErrNotExist) {
 		return GpuVendorTenstorrent
 	}
+	if _, err := os.Stat("/dev/dxg"); !errors.Is(err, os.ErrNotExist) {
+		// WSL2
+		if _, err := os.Stat("/usr/lib/wsl/lib/nvidia-smi"); !errors.Is(err, os.ErrNotExist) {
+			return GpuVendorNvidia
+		}
+	}
 	return GpuVendorNone
 }