Unverified Commit 6d283882 authored by Bruce MacDonald's avatar Bruce MacDonald Committed by GitHub
Browse files

catch insufficient permissions nvidia err (#934)

parent 5c3491f4
......@@ -212,6 +212,10 @@ func CheckVRAM() (int64, error) {
scanner := bufio.NewScanner(&stdout)
for scanner.Scan() {
line := scanner.Text()
if strings.Contains(line, "[Insufficient Permissions]") {
return 0, fmt.Errorf("GPU support may not enabled, check you have installed GPU drivers and have the necessary permissions to run nvidia-smi")
}
vram, err := strconv.ParseInt(strings.TrimSpace(line), 10, 64)
if err != nil {
return 0, fmt.Errorf("failed to parse available VRAM: %v", err)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment