mirror of
https://github.com/NVIDIA/OpenShell.git
synced 2026-10-17 05:31:16 +08:00
Evan Lezar
d39c1bc603
On Jetson/Tegra platforms nvidia-smi is installed at /usr/sbin/nvidia-smi rather than /usr/bin/nvidia-smi and may not be on PATH inside the sandbox. Fall back to the full path when the bare command is not found. Signed-off-by: Evan Lezar <elezar@nvidia.com>
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
|
|
from openshell._proto import datamodel_pb2
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import Callable
|
|
|
|
from openshell import Sandbox
|
|
|
|
|
|
@pytest.mark.gpu
|
|
def test_gpu_sandbox_reports_available_gpu(
|
|
sandbox: Callable[..., Sandbox],
|
|
gpu_sandbox_spec: datamodel_pb2.SandboxSpec,
|
|
) -> None:
|
|
nvidia_smi_args = ["--query-gpu=name", "--format=csv,noheader"]
|
|
with sandbox(spec=gpu_sandbox_spec, delete_on_exit=True) as sb:
|
|
result = sb.exec(["nvidia-smi", *nvidia_smi_args], timeout_seconds=30)
|
|
if result.exit_code != 0:
|
|
# On some platforms (e.g. Tegra/Jetson) nvidia-smi lives in
|
|
# /usr/sbin rather than /usr/bin and may not be on PATH.
|
|
result = sb.exec(["/usr/sbin/nvidia-smi", *nvidia_smi_args], timeout_seconds=30)
|
|
|
|
assert result.exit_code == 0, result.stderr
|
|
assert result.stdout.strip()
|