Documentation ¶
Index ¶
- Variables
- func BaseDockerImageName(projectDir string) string
- func CUDABaseImageFor(cuda string, cuDNN string) (string, error)
- func DockerImageName(projectDir string) string
- func GetProjectDir(customDir string) (string, error)
- func Validate(yamlConfig string, version string) error
- func ValidateConfig(config *Config, version string) error
- func ValidateSchema(schemaLoader, dataLoader gojsonschema.JSONLoader) error
- type ArgumentType
- type Build
- type CUDABaseImage
- type Config
- type Example
- type Image
- type RunArgument
- type Stats
- type TFCompatibility
- type TorchCompatibility
- type Version
Constants ¶
This section is empty.
Variables ¶
View Source
var CUDABaseImages []CUDABaseImage
View Source
var TFCompatibilityMatrix []TFCompatibility
View Source
var TorchCompatibilityMatrix []TorchCompatibility
Functions ¶
func BaseDockerImageName ¶
BaseDockerImageName returns the Docker image name for base images
func DockerImageName ¶
DockerImageName returns the default Docker image name for images
func GetProjectDir ¶
Returns the project's root directory, or the directory specified by the --project-dir flag
func ValidateConfig ¶
func ValidateSchema ¶
func ValidateSchema(schemaLoader, dataLoader gojsonschema.JSONLoader) error
Types ¶
type ArgumentType ¶
type ArgumentType string
const ( ArgumentTypeString ArgumentType = "str" ArgumentTypeInt ArgumentType = "int" ArgumentTypeFloat ArgumentType = "float" ArgumentTypeBool ArgumentType = "bool" ArgumentTypePath ArgumentType = "Path" )
type Build ¶
type Build struct { GPU bool `json:"gpu,omitempty" yaml:"gpu"` PythonVersion string `json:"python_version,omitempty" yaml:"python_version"` PythonRequirements string `json:"python_requirements,omitempty" yaml:"python_requirements"` PythonPackages []string `json:"python_packages,omitempty" yaml:"python_packages"` // Deprecated, but included for backwards compatibility Run []string `json:"run,omitempty" yaml:"run"` SystemPackages []string `json:"system_packages,omitempty" yaml:"system_packages"` PreInstall []string `json:"pre_install,omitempty" yaml:"pre_install"` // Deprecated, but included for backwards compatibility CUDA string `json:"cuda,omitempty" yaml:"cuda"` CuDNN string `json:"cudnn,omitempty" yaml:"cudnn"` // contains filtered or unexported fields }
type CUDABaseImage ¶
func (*CUDABaseImage) ImageTag ¶
func (i *CUDABaseImage) ImageTag() string
func (*CUDABaseImage) UnmarshalJSON ¶
func (i *CUDABaseImage) UnmarshalJSON(data []byte) error
type Config ¶
type Config struct { Build *Build `json:"build" yaml:"build"` Image string `json:"image,omitempty" yaml:"image"` Predict string `json:"predict,omitempty" yaml:"predict"` Train string `json:"train,omitempty" yaml:"train"` }
func DefaultConfig ¶
func DefaultConfig() *Config
func GetConfig ¶
Loads and instantiates a Config object customDir can be specified to override the default - current working directory
func (*Config) CUDABaseImageTag ¶
func (*Config) PythonRequirementsForArch ¶ added in v0.0.13
PythonRequirementsForArch returns a requirements.txt file with all the GPU packages resolved for given OS and architecture.
func (*Config) ValidateAndComplete ¶ added in v0.0.13
func (*Config) ValidateAndCompleteCUDA ¶ added in v0.0.6
type RunArgument ¶
type TFCompatibility ¶
type TFCompatibility struct { TF string TFCPUPackage string TFGPUPackage string CUDA string CuDNN string Pythons []string }
func (*TFCompatibility) UnmarshalJSON ¶
func (compat *TFCompatibility) UnmarshalJSON(data []byte) error
type TorchCompatibility ¶
type TorchCompatibility struct { Torch string Torchvision string Torchaudio string FindLinks string ExtraIndexURL string CUDA *string Pythons []string }
func (*TorchCompatibility) TorchVersion ¶
func (c *TorchCompatibility) TorchVersion() string
func (*TorchCompatibility) TorchvisionVersion ¶
func (c *TorchCompatibility) TorchvisionVersion() string
Click to show internal directories.
Click to hide internal directories.