Skip to content

nvidia_tasks

detect_cuda()

High-level CUDA detection.

Simple rule: CUDA is considered present if nvcc is available.

Source code in src/awesome_os/tasks/system/nvidia_tasks.py 
32
33
34
35
36
37
38
39
40
41
42
43
44
45
def detect_cuda() -> TaskResult:
    """High-level CUDA detection.

    Simple rule: CUDA is considered present if `nvcc` is available.
    """
    system = platform.system().lower()
    if not _command_exists("nvcc"):
        return TaskResult(ok=False, summary=f"CUDA not detected (nvcc missing) ({system})")

    res = run(["nvcc", "--version"], check=False)
    details = (res.stdout + "\n" + res.stderr).strip()
    if res.returncode == 0:
        return TaskResult(ok=True, summary="CUDA detected (nvcc available)", details=details)
    return TaskResult(ok=False, summary="CUDA not detected (nvcc failed)", details=details)

detect_nvidia()

Detect NVIDIA driver availability via nvidia-smi.

Works on Linux and Windows (including CUDA on Windows).

Source code in src/awesome_os/tasks/system/nvidia_tasks.py 
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
def detect_nvidia() -> TaskResult:
    """Detect NVIDIA driver availability via `nvidia-smi`.

    Works on Linux and Windows (including CUDA on Windows).
    """
    if not _command_exists("nvidia-smi"):
        system = platform.system().lower()
        return TaskResult(ok=False, summary=f"nvidia-smi not found on PATH ({system})")

    # Keep it simple: run a lightweight query.
    res = run(["nvidia-smi", "-L"], check=False)
    details = (res.stdout + "\n" + res.stderr).strip()
    if res.returncode == 0:
        return TaskResult(ok=True, summary="nvidia-smi: ok", details=details)
    return TaskResult(ok=False, summary="nvidia-smi: failed", details=details)

setup_cuda()

Advanced: CUDA setup task (placeholder).

Keeping it simple for now: report-only stub.

Source code in src/awesome_os/tasks/system/nvidia_tasks.py 
48
49
50
51
52
53
54
55
56
57
58
def setup_cuda() -> TaskResult:
    """Advanced: CUDA setup task (placeholder).

    Keeping it simple for now: report-only stub.
    """
    system = platform.system().lower()
    return TaskResult(
        ok=False,
        summary=f"CUDA setup (advanced) not implemented yet ({system})",
        details="Planned: provide guided installation steps for Windows/Linux, then validate with nvcc and nvidia-smi.",
    )

setup_nvidia_ubuntu()

Setup NVIDIA driver on Ubuntu.

Source code in src/awesome_os/tasks/system/nvidia_tasks.py 
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
def setup_nvidia_ubuntu() -> TaskResult:
    """Setup NVIDIA driver on Ubuntu."""
    if _is_wsl():
        return setup_nvidia_wsl_instructions()

    ubuntu_drivers = shutil.which("ubuntu-drivers")
    if ubuntu_drivers is None:
        return TaskResult(
            ok=False,
            summary="ubuntu-drivers not found",
            details=(
                "Recommended (Ubuntu): install it first, then re-run:\n"
                "sudo apt-get update\n"
                "sudo apt-get install -y ubuntu-drivers-common"
            ),
        )

    actions: list[str] = []

    actions.append("Planned steps (native Linux):")
    actions.append("1) sudo ubuntu-drivers devices")
    actions.append("2) sudo ubuntu-drivers autoinstall")
    actions.append("3) reboot")
    actions.append("4) verify: nvidia-smi")

    devices = run([ubuntu_drivers, "devices"], check=False)
    devices_out = (devices.stdout + "\n" + devices.stderr).strip()
    if devices_out:
        actions.append("\nubuntu-drivers devices output:")
        actions.append(devices_out)

    cmd = ["sudo", ubuntu_drivers, "autoinstall"]
    res = run(cmd, check=False)
    details = (res.stdout + "\n" + res.stderr).strip()
    actions.append(f"\nran: {' '.join(cmd)}")
    if details:
        actions.append(details)

    if res.returncode == 0:
        actions.append("\nDriver install complete. Please reboot, then run: nvidia-smi")
        return TaskResult(
            ok=True,
            summary="NVIDIA driver install: completed (reboot required)",
            details="\n".join(actions).strip(),
        )

    if "a password is required" in details.lower() or "sudo:" in details.lower():
        actions.append(
            "\nsudo needs a password. Run manually in a terminal:\nsudo ubuntu-drivers autoinstall"
        )
        return TaskResult(
            ok=False,
            summary="NVIDIA install requires sudo password",
            details="\n".join(actions).strip(),
        )

    return TaskResult(
        ok=False, summary="NVIDIA driver install: failed", details="\n".join(actions).strip()
    )

setup_nvidia_windows()

Setup NVIDIA driver on Windows.

Source code in src/awesome_os/tasks/system/nvidia_tasks.py 
139
140
141
def setup_nvidia_windows() -> TaskResult:
    """Setup NVIDIA driver on Windows."""
    return TaskResult(ok=False, summary="NVIDIA setup for Windows not implemented yet")

setup_nvidia_wsl_instructions()

WSL: provide instructions for installing NVIDIA drivers on the Windows host.

Source code in src/awesome_os/tasks/system/nvidia_tasks.py 
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
def setup_nvidia_wsl_instructions() -> TaskResult:
    """WSL: provide instructions for installing NVIDIA drivers on the Windows host."""
    return TaskResult(
        ok=True,
        summary="NVIDIA setup: WSL detected",
        details=(
            "You are running inside WSL. Install/update the NVIDIA driver on Windows (host), then restart WSL.\n"
            "Steps:\n"
            "1) Install latest NVIDIA Windows driver from https://www.nvidia.com/en-us/drivers/\n"
            "2) Reboot Windows\n"
            "3) In PowerShell: nvidia-smi\n"
            "4) In PowerShell: wsl --shutdown\n"
            "5) Re-open your Ubuntu/WSL terminal and run: nvidia-smi"
        ),
    )