ir

Documentation

Index

• Variables
• func Base(os, language string) error
• func CUDA(version, cudnn string)
• func Compile(ctx context.Context, cachePrefix string, pub string) (*llb.Definition, error)
• func CondaChannel(channel string) error
• func CondaPackage(deps []string, channel []string)
• func GPU(numGPUs int)
• func GPUEnabled() bool
• func Git(name, email, editor string) error
• func Jupyter(pwd string, port int64) error
• func Labels() (map[string]string, error)
• func NumGPUs() int
• func PyPIIndex(mode, url, extraURL string) error
• func PyPIPackage(deps []string)
• func RPackage(deps []string)
• func Run(commands []string) error
• func Shell(shell string) error
• func SystemPackage(deps []string)
• func UbuntuAPT(mode, source string) error
• func VSCodePlugins(plugins []string) error
• type CondaConfig
• type GitConfig
• type Graph
  • func NewGraph() *Graph
  • func (g Graph) CacheID(filename string) string
  • func (g Graph) Compile(uid, gid int) (llb.State, error)
  • func (g Graph) CondaEnabled() bool
  • func (g Graph) GPUEnabled() bool
  • func (g Graph) Labels() (map[string]string, error)
• type JupyterConfig
• type Language

Variables

var DefaultGraph = NewGraph()

Functions

func Base

func Base(os, language string) error

func CUDA

func CUDA(version, cudnn string)

func Compile

func Compile(ctx context.Context, cachePrefix string, pub string) (*llb.Definition, error)

func CondaChannel

func CondaChannel(channel string) error

func CondaPackage

func CondaPackage(deps []string, channel []string)

func GPU

func GPU(numGPUs int)

func GPUEnabled

func GPUEnabled() bool

func Git

func Git(name, email, editor string) error

func Jupyter

func Jupyter(pwd string, port int64) error

func Labels

func Labels() (map[string]string, error)

func NumGPUs

func NumGPUs() int

func PyPIIndex

func PyPIIndex(mode, url, extraURL string) error

func PyPIPackage

func PyPIPackage(deps []string)

func RPackage

func RPackage(deps []string)

func Run

func Run(commands []string) error

func Shell

func Shell(shell string) error

func SystemPackage

func SystemPackage(deps []string)

func UbuntuAPT

func UbuntuAPT(mode, source string) error

UbuntuAPT updates the Ubuntu apt source.list in the image.

func VSCodePlugins

func VSCodePlugins(plugins []string) error

Types

type CondaConfig

type CondaConfig struct {
	CondaPackages      []string
	AdditionalChannels []string
	CondaChannel       *string
}

type GitConfig

type GitConfig struct {
	Name   string
	Email  string
	Editor string
}

type Graph

type Graph struct {
	OS string
	Language

	Shell   string
	CUDA    *string
	CUDNN   *string
	NumGPUs int

	UbuntuAPTSource   *string
	PyPIIndexURL      *string
	PyPIExtraIndexURL *string

	PublicKeyPath string

	PyPIPackages   []string
	RPackages      []string
	SystemPackages []string

	VSCodePlugins []vscode.Plugin

	Exec []string
	*JupyterConfig
	*GitConfig
	*CondaConfig

	Writer      compileui.Writer
	CachePrefix string
	// contains filtered or unexported fields
}

A Graph contains the state, such as its call stack and thread-local storage.

func NewGraph

func NewGraph() *Graph

func (Graph) CacheID

func (g Graph) CacheID(filename string) string

func (Graph) Compile

func (g Graph) Compile(uid, gid int) (llb.State, error)

func (Graph) CondaEnabled

func (g Graph) CondaEnabled() bool

func (Graph) GPUEnabled

func (g Graph) GPUEnabled() bool

func (Graph) Labels

func (g Graph) Labels() (map[string]string, error)

type JupyterConfig

type JupyterConfig struct {
	Password string
	Port     int64
}

type Language

type Language struct {
	Name    string
	Version *string
}