nvidia

Documentation

Index

• Constants
• Variables
• func Copy(src, dst string) error
• func GetCUDAVersion() (string, error)
• func GetControlDevicePaths() ([]string, error)
• func GetDriverVersion() (string, error)
• func Init() error
• func LoadUVM() error
• func Shutdown() error
• type CUDADevice
• type Device
  • func FilterDevices(devs []Device, ids []string) ([]Device, error)
  • func LookupDevices(s ...LookupStrategy) (devs []Device, err error)
  • func (d *Device) Status() (*DeviceStatus, error)
• type DeviceStatus
• type FileCloneStrategy
• type LinkOrCopyStrategy
  • func (s LinkOrCopyStrategy) Clone(src, dst string) error
• type LinkStrategy
  • func (s LinkStrategy) Clone(src, dst string) error
• type LookupStrategy
• type NVMLDevice
• type NVMLDeviceStatus
• type Volume
  • func (v *Volume) Create(s FileCloneStrategy) (err error)
  • func (v *Volume) Exists(version ...string) (bool, error)
  • func (v *Volume) ListVersions() ([]string, error)
  • func (v *Volume) Remove(version ...string) error
• type VolumeInfo
• type VolumeMap
  • func LookupVolumes(prefix string) (vols VolumeMap, err error)

Constants

const (
	DockerPlugin   = "nvidia-docker"
	DeviceCtl      = "/dev/nvidiactl"
	DeviceUVM      = "/dev/nvidia-uvm"
	DeviceUVMTools = "/dev/nvidia-uvm-tools"
)

Variables

var Volumes = []VolumeInfo{
	{
		"nvidia_driver",
		"/usr/local/nvidia",
		"ro",
		components{
			"binaries": {

				"nvidia-cuda-mps-control",
				"nvidia-cuda-mps-server",
				"nvidia-debugdump",
				"nvidia-persistenced",
				"nvidia-smi",
			},
			"libraries": {

				"libnvidia-ml.so",
				"libcuda.so",
				"libnvidia-ptxjitcompiler.so",
				"libnvidia-fatbinaryloader.so",
				"libnvidia-opencl.so",
				"libnvidia-compiler.so",

				"libvdpau_nvidia.so",
				"libnvidia-encode.so",
				"libnvcuvid.so",
				"libnvidia-fbc.so",
				"libnvidia-ifr.so",

				"libGL.so",
				"libGLX.so",
				"libOpenGL.so",
				"libGLESv1_CM.so",
				"libGLESv2.so",
				"libEGL.so",
				"libGLdispatch.so",

				"libGLX_nvidia.so",
				"libEGL_nvidia.so",
				"libGLESv2_nvidia.so",
				"libGLESv1_CM_nvidia.so",
				"libnvidia-eglcore.so",
				"libnvidia-egl-wayland.so",
				"libnvidia-glcore.so",
				"libnvidia-tls.so",
				"libnvidia-glsi.so",
			},
		},
	},
}

Functions

func Copy

func Copy(src, dst string) error

func GetCUDAVersion

func GetCUDAVersion() (string, error)

func GetControlDevicePaths

func GetControlDevicePaths() ([]string, error)

func GetDriverVersion

func GetDriverVersion() (string, error)

func Init

func Init() error

func LoadUVM

func LoadUVM() error

func Shutdown

func Shutdown() error

Types

type CUDADevice

type CUDADevice cuda.Device

type Device

type Device struct {
	*NVMLDevice
	*CUDADevice
}

func FilterDevices

func FilterDevices(devs []Device, ids []string) ([]Device, error)

func LookupDevices

func LookupDevices(s ...LookupStrategy) (devs []Device, err error)

func (*Device) Status

func (d *Device) Status() (*DeviceStatus, error)

type DeviceStatus

type DeviceStatus struct {
	*NVMLDeviceStatus
}

type FileCloneStrategy

type FileCloneStrategy interface {
	Clone(src, dst string) error
}

type LinkOrCopyStrategy

type LinkOrCopyStrategy struct{}

func (LinkOrCopyStrategy) Clone

func (s LinkOrCopyStrategy) Clone(src, dst string) error

type LinkStrategy

type LinkStrategy struct{}

func (LinkStrategy) Clone

func (s LinkStrategy) Clone(src, dst string) error

type LookupStrategy

type LookupStrategy uint

const (
	LookupMinimal LookupStrategy = iota
)

type NVMLDevice

type NVMLDevice nvml.Device

type NVMLDeviceStatus

type NVMLDeviceStatus nvml.DeviceStatus

type Volume

type Volume struct {
	*VolumeInfo

	Path    string
	Version string
	// contains filtered or unexported fields
}

func (*Volume) Create

func (v *Volume) Create(s FileCloneStrategy) (err error)

func (*Volume) Exists

func (v *Volume) Exists(version ...string) (bool, error)

func (*Volume) ListVersions

func (v *Volume) ListVersions() ([]string, error)

func (*Volume) Remove

func (v *Volume) Remove(version ...string) error

type VolumeInfo

type VolumeInfo struct {
	Name         string
	Mountpoint   string
	MountOptions string
	Components   components
}

type VolumeMap

type VolumeMap map[string]*Volume

func LookupVolumes

func LookupVolumes(prefix string) (vols VolumeMap, err error)