# Copyright 2017 Google Inc. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. apiVersion: extensions/v1beta1 kind: DaemonSet metadata: name: nvidia-driver-installer namespace: kube-system spec: template: metadata: labels: name: nvidia-driver-installer spec: tolerations: - key: "nvidia.com/gpu" effect: "NoSchedule" operator: "Exists" hostNetwork: true hostPID: true volumes: - name: dev hostPath: path: /dev - name: nvidia-install-dir-host hostPath: path: /home/kubernetes/bin/nvidia - name: root-mount hostPath: path: / initContainers: - image: gcr.io/google-containers/ubuntu-nvidia-driver-installer@sha256:7ffaf40fcf6bcc5bc87501b6be295a47ce74e1f7aac914a9f3e6c6fb8dd780a4 name: nvidia-driver-installer resources: requests: cpu: 1 limits: cpu: 1 securityContext: privileged: true env: - name: NVIDIA_INSTALL_DIR_HOST value: /home/kubernetes/bin/nvidia - name: NVIDIA_INSTALL_DIR_CONTAINER value: /usr/local/nvidia - name: ROOT_MOUNT_DIR value: /root # This driver supports CUDA 8 and a Tesla K80 card # - name: NVIDIA_DRIVER_VERSION # value: "384.66" # # This driver supports CUDA 9 and a Tesla K80 card # # Does not work well on linux-image-4.4.0-116-generic # See https://bugs.launchpad.net/ubuntu/+source/xorg/+bug/1750937 - name: NVIDIA_DRIVER_VERSION value: "384.111" volumeMounts: - name: nvidia-install-dir-host mountPath: /usr/local/nvidia - name: dev mountPath: /dev - name: root-mount mountPath: /root containers: - image: "gcr.io/google_containers/pause-amd64:3.0" name: pause