Documentation
¶
Index ¶
- Variables
- func ExtractSubFeaturesFromPytorchVersion(pytorchVersion string) (string, string, string, string, string, error)
- func FetchCUDABaseImages() ([]config.CUDABaseImage, error)
- func FetchTensorFlowCompatibilityMatrix() ([]config.TFCompatibility, error)
- func FetchTorchCompatibilityMatrix() ([]config.TorchCompatibility, error)
- func FindCompatiblePythonVersions(torchVersion string, torchVisionVersion string, torchAudioVersion string, ...) ([]string, error)
- type TorchPackage
Constants ¶
This section is empty.
Variables ¶
View Source
var ErrorBadPytorchFormat = errors.New("The pytorch version format could not be parsed.")
Functions ¶
func ExtractSubFeaturesFromPytorchVersion ¶ added in v0.15.11
func FetchCUDABaseImages ¶
func FetchCUDABaseImages() ([]config.CUDABaseImage, error)
func FetchTensorFlowCompatibilityMatrix ¶
func FetchTensorFlowCompatibilityMatrix() ([]config.TFCompatibility, error)
func FetchTorchCompatibilityMatrix ¶
func FetchTorchCompatibilityMatrix() ([]config.TorchCompatibility, error)
Types ¶
type TorchPackage ¶ added in v0.15.11
type TorchPackage struct {
Name string
Version string
Variant string
CUDA *string
PythonVersion string
}
func FetchTorchPackages ¶ added in v0.15.11
func FetchTorchPackages(name string) ([]TorchPackage, error)
func (*TorchPackage) Equals ¶ added in v0.16.2
func (c *TorchPackage) Equals(other TorchPackage) bool
Source Files
Click to show internal directories.
Click to hide internal directories.