hardware

Documentation

Index

• func DetectMemory()
• type GPU
  • func DetectProcessingDevice() GPU
  • func (i GPU) String() string

Functions

func DetectMemory

func DetectMemory()

DetectMemory detects the memory of the system TODO: Use to determine if we can run certain AI models

Types

type GPU

type GPU int

GPU Processing devices used for AI inference

const (
	AMD    GPU = iota // AMD
	NVIDIA            // Nvidia

)

func DetectProcessingDevice

func DetectProcessingDevice() GPU

func (GPU) String

func (i GPU) String() string