Merge pull request #55737 from mindprince/update-nvidia-urls

Automatic merge from submit-queue. If you want to cherry-pick this change to another branch, please follow the instructions <a href="https://github.com/kubernetes/community/blob/master/contributors/devel/cherry-picks.md">here</a>.

Update URLs for nvidia gpu device plugin and nvidia driver installer.

Device plugin is now an addon and its manifest is now in kubernetes/kubernetes. The manifest on
GoogleCloudPlatform/container-engine-accelerators no longer contains device plugin.

This is needed after https://github.com/kubernetes/kubernetes/pull/54826 and https://github.com/GoogleCloudPlatform/container-engine-accelerators/pull/25

**Release note**:
```release-note
NONE
```

/sig scheduling
This commit is contained in:
Kubernetes Submit Queue 2017-11-18 09:36:05 -08:00 committed by GitHub
commit 2d972c19bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions

View File

@ -32,7 +32,7 @@ const (
// TODO: Parametrize it by making it a feature in TestFramework. // TODO: Parametrize it by making it a feature in TestFramework.
// so we can override the daemonset in other setups (non COS). // so we can override the daemonset in other setups (non COS).
// GPUDevicePluginDSYAML is the official Google Device Plugin Daemonset NVIDIA GPU manifest for GKE // GPUDevicePluginDSYAML is the official Google Device Plugin Daemonset NVIDIA GPU manifest for GKE
GPUDevicePluginDSYAML = "https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/master/device-plugin-daemonset.yaml" GPUDevicePluginDSYAML = "https://raw.githubusercontent.com/kubernetes/kubernetes/master/cluster/addons/device-plugins/nvidia-gpu/daemonset.yaml"
) )
// TODO make this generic and not linked to COS only // TODO make this generic and not linked to COS only
@ -61,8 +61,8 @@ func NVIDIADevicePlugin(ns string) *v1.Pod {
Spec: ds.Spec.Template.Spec, Spec: ds.Spec.Template.Spec,
} }
// Remove NVIDIA drivers installation // Remove node affinity
p.Spec.InitContainers = []v1.Container{} p.Spec.Affinity = nil
return p return p
} }

View File

@ -163,7 +163,7 @@ func testNvidiaGPUsOnCOS(f *framework.Framework) {
framework.Logf("Cluster is running on COS. Proceeding with test") framework.Logf("Cluster is running on COS. Proceeding with test")
if f.BaseName == "device-plugin-gpus" { if f.BaseName == "device-plugin-gpus" {
dsYamlUrl = framework.GPUDevicePluginDSYAML dsYamlUrl = "https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/master/daemonset.yaml"
gpuResourceName = framework.NVIDIAGPUResourceName gpuResourceName = framework.NVIDIAGPUResourceName
podCreationFunc = makeCudaAdditionDevicePluginTestPod podCreationFunc = makeCudaAdditionDevicePluginTestPod
} else { } else {