nvidia

package

v0.0.0-...-e0141d8 Latest Latest Go to latest Published: Oct 3, 2017 License: BSD-3-Clause Imports: 18 Imported by: 2

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

Repository

github.com/KamranAlipour/nvidia-docker

Links

Open Source Insights

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 ¶

View Source

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

Variables ¶

View Source

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)

Source Files ¶

View all Source files

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