ir

Documentation

Index

• Variables
• func Base(os, language string)
• func CUDA(version, cudnn string)
• func Compile(ctx context.Context) (*llb.Definition, error)
• func GPUEnabled() bool
• func Jupyter(pwd string, port int64) error
• func PyPIMirror(mode, mirror string) error
• func PyPIPackage(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 Graph
  • func NewGraph() *Graph
  • func (g Graph) Compile() (llb.State, error)
• type JupyterConfig

Variables

var DefaultGraph = NewGraph()

Functions

func Base

func Base(os, language string)

func CUDA

func CUDA(version, cudnn string)

func Compile

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

func GPUEnabled

func GPUEnabled() bool

func Jupyter

func Jupyter(pwd string, port int64) error

func PyPIMirror

func PyPIMirror(mode, mirror string) error

func PyPIPackage

func PyPIPackage(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 Graph

type Graph struct {
	OS       string
	Language string
	Shell    string
	CUDA     *string
	CUDNN    *string

	UbuntuAPTSource *string
	PyPIMirror      *string

	BuiltinSystemPackages []string
	PyPIPackages          []string
	SystemPackages        []string
	VSCodePlugins         []vscode.Plugin

	Exec []string
	*JupyterConfig

	Writer compileui.Writer
}

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

func NewGraph

func NewGraph() *Graph

func (Graph) Compile

func (g Graph) Compile() (llb.State, error)

type JupyterConfig

type JupyterConfig struct {
	Password string
	Port     int64
}