Documentation
¶
Index ¶
- func HasCANNRuntime(ctx context.Context, dockerClient client.SystemAPIClient) (bool, error)
- func HasMTHREADSRuntime(ctx context.Context, dockerClient client.SystemAPIClient) (bool, error)
- func HasNVIDIARuntime(ctx context.Context, dockerClient client.SystemAPIClient) (bool, error)
- func HasROCmRuntime(ctx context.Context, dockerClient client.SystemAPIClient) (bool, error)
- type GPUSupport
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func HasCANNRuntime ¶ added in v1.0.0
HasCANNRuntime determines whether there is a Ascend CANN runtime available
func HasMTHREADSRuntime ¶ added in v1.0.0
HasMTHREADSRuntime determines whether there is a mthreads runtime available
func HasNVIDIARuntime ¶
HasNVIDIARuntime determines whether there is an nvidia runtime available
func HasROCmRuntime ¶ added in v0.1.46
HasROCmRuntime determines whether there is a ROCm 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 // GPUSupportROCm indicates ROCm GPU support. GPUSupportROCm // GPUSupportMUSA indicates MUSA GPU support. GPUSupportMUSA // GPUSupportCANN indicates Ascend NPU support. GPUSupportCANN )
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
Click to show internal directories.
Click to hide internal directories.