2016-01-06 00:45:23 +01:00
|
|
|
// Copyright (c) 2015-2016, NVIDIA CORPORATION. All rights reserved.
|
2015-12-11 09:14:15 +01:00
|
|
|
|
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"strings"
|
|
|
|
|
2017-02-06 22:39:25 +01:00
|
|
|
"github.com/NVIDIA/nvidia-docker/src/docker"
|
2015-12-11 09:14:15 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
labelCUDAVersion = "com.nvidia.cuda.version"
|
|
|
|
labelVolumesNeeded = "com.nvidia.volumes.needed"
|
|
|
|
)
|
|
|
|
|
2015-12-17 00:36:17 +01:00
|
|
|
func VolumesNeeded(image string) ([]string, error) {
|
2016-01-12 02:25:56 +01:00
|
|
|
ok, err := docker.ImageExists(image)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if !ok {
|
|
|
|
if err = docker.ImagePull(image); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-12-17 00:36:17 +01:00
|
|
|
label, err := docker.Label(image, labelVolumesNeeded)
|
2015-12-12 11:35:07 +01:00
|
|
|
if err != nil {
|
2015-12-17 00:36:17 +01:00
|
|
|
return nil, err
|
2015-12-12 11:35:07 +01:00
|
|
|
}
|
2015-12-17 00:36:17 +01:00
|
|
|
if label == "" {
|
|
|
|
return nil, nil
|
2015-12-12 11:35:07 +01:00
|
|
|
}
|
2015-12-17 00:36:17 +01:00
|
|
|
return strings.Split(label, " "), nil
|
2015-12-12 11:35:07 +01:00
|
|
|
}
|
|
|
|
|
2015-12-17 00:36:17 +01:00
|
|
|
func cudaSupported(image, version string) error {
|
2015-12-11 09:14:15 +01:00
|
|
|
var vmaj, vmin int
|
|
|
|
var lmaj, lmin int
|
|
|
|
|
2015-12-12 04:02:13 +01:00
|
|
|
label, err := docker.Label(image, labelCUDAVersion)
|
2015-12-11 09:14:15 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if label == "" {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
if _, err := fmt.Sscanf(version, "%d.%d", &vmaj, &vmin); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if _, err := fmt.Sscanf(label, "%d.%d", &lmaj, &lmin); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if lmaj > vmaj || (lmaj == vmaj && lmin > vmin) {
|
2016-04-16 00:13:49 +02:00
|
|
|
return fmt.Errorf("unsupported CUDA version: driver %s < image %s", version, label)
|
2015-12-11 09:14:15 +01:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|