gpu

package

v0.1.45 Latest Latest Go to latest Published: Oct 17, 2025 License: Apache-2.0 Imports: 3 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/docker/model-runner

Links

Open Source Insights

Documentation ¶

Index ¶

func HasNVIDIARuntime(ctx context.Context, dockerClient client.SystemAPIClient) (bool, error)
type GPUSupport
- func ProbeGPUSupport(ctx context.Context, dockerClient client.SystemAPIClient) (GPUSupport, error)

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func HasNVIDIARuntime ¶

func HasNVIDIARuntime(ctx context.Context, dockerClient client.SystemAPIClient) (bool, error)

HasNVIDIARuntime determines whether there is an nvidia runtime available

Types ¶

type GPUSupport ¶

type GPUSupport uint8

GPUSupport encodes the GPU support available on a Docker engine.

const (
	// GPUSupportNone indicates no detectable GPU support.
	GPUSupportNone GPUSupport = iota
	// GPUSupportCUDA indicates CUDA GPU support.
	GPUSupportCUDA
)

func ProbeGPUSupport ¶

func ProbeGPUSupport(ctx context.Context, dockerClient client.SystemAPIClient) (GPUSupport, error)

ProbeGPUSupport determines whether or not the Docker engine has GPU support.

Source Files ¶

View all Source files

gpu.go

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL