in partition_gpu/partition_gpu.go [289:313]
func createGPUPartitions(partitionSize string) error {
p, err := buildPartitionStr(partitionSize)
if err != nil {
return err
}
args := []string{"mig", "-cgi", p}
glog.Infof("Running %s %s", *nvidiaSmiPath, strings.Join(args, " "))
out, err := exec.Command(*nvidiaSmiPath, args...).Output()
if err != nil {
return fmt.Errorf("failed to create GPU Instances: output: %s, error: %v", string(out), err)
}
glog.Infof("Output:\n %s", string(out))
args = []string{"mig", "-cci"}
glog.Infof("Running %s %s", *nvidiaSmiPath, strings.Join(args, " "))
out, err = exec.Command(*nvidiaSmiPath, args...).Output()
if err != nil {
return fmt.Errorf("failed to create compute instances: output: %s, error: %v", string(out), err)
}
glog.Infof("Output:\n %s", string(out))
return nil
}