配置gpu节点的runtime

vim /etc/docker/daemon.json
{
    "default-runtime": "nvidia",
    "runtimes": {
        "nvidia": {
            "path": "/usr/bin/nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}

安装插件

前提：

Kubernetes nodes have to be pre-installed with NVIDIA drivers.
Kubernetes nodes have to be pre-installed with nvidia-docker 2.0
nvidia-container-runtime must be configured as the default runtime for docker instead of runc.
NVIDIA drivers ~= 361.93

kubectl create -f https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/v1.10/nvidia-device-plugin.yml
kubectl get pods -n kube-system

等pods启动后会发现所有节点都有nvidia-device-plugin这个container在运行。

参考:

K8S GPU支持
Schedule GPUs
github NVIDIA/k8s-device-plugin