| #!/usr/bin/env bash |
| |
| # Copyright 2017 The Kubernetes Authors. |
| # |
| # 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. |
| |
| # This script is meant to install Nvidia drivers on Ubuntu 14.04 GCE VMs. |
| # This script is meant to facilitate testing of Nvidia GPU support in Kubernetes. |
| echo "Checking for CUDA and installing." |
| # Check for CUDA and try to install. |
| if ! dpkg-query -W cuda; then |
| curl -O http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.61-1_amd64.deb |
| dpkg -i ./cuda-repo-ubuntu1404_8.0.61-1_amd64.deb |
| apt-get update |
| apt-get install cuda -y |
| apt-get install linux-headers-$(uname -r) -y |
| fi |
| ## Pre-loads kernel modules |
| nvidia-smi |