MCPcopy Index your code
hub / github.com/coder/envbox / assertInnerNvidiaSMI

Function assertInnerNvidiaSMI

integration/gpu_test.go:199–205  ·  view source on GitHub ↗

assertInnerNvidiaSMI checks that nvidia-smi runs successfully in the inner container.

(ctx context.Context, t *testing.T, containerID string)

Source from the content-addressed store, hash-verified

197// assertInnerNvidiaSMI checks that nvidia-smi runs successfully in the inner
198// container.
199func assertInnerNvidiaSMI(ctx context.Context, t *testing.T, containerID string) {
200 t.Helper()
201 // Assert that we can run nvidia-smi in the inner container.
202 out, err := execContainerCmd(ctx, t, containerID, "docker", "exec", "workspace_cvm", "nvidia-smi")
203 require.NoError(t, err, "failed to run nvidia-smi in the inner container")
204 require.Contains(t, out, "NVIDIA-SMI", "nvidia-smi output does not contain NVIDIA-SMI")
205}
206
207// startEnvboxCmd starts the envbox container with the given arguments.
208// Ideally we would use ory/dockertest for this, but it doesn't support

Callers 1

TestDocker_NvidiaFunction · 0.85

Calls 1

execContainerCmdFunction · 0.85

Tested by

no test coverage detected