Kubernetes 配置节点支持 GPU

本文以 Debian 10, kubernetes 1.19.3 为例，进行配置 GPU 支持。

参考资料：

• nvidia-runtime install
• docker 添加 nvidia 运行环境
• Schedule GPUs
• https://github.com/NVIDIA/k8s-device-plugin
• https://www.kubeflow.org/docs/notebooks/setup/

1. 前提条件

• 节点上已经安装了 nvidia 显卡驱动。驱动版本必须大于 384.81 。
• 节点上已经安装了 nvidia-docker 2.0 插件。安装过程见 docker 添加 nvidia 运行环境 。
• nvidia-container-runtime 作为默认的 runtime 。 即在 /etc/docker/daemon.json 中设置 "default-runtime": "nvidia"

2. 安装插件

1

kubectl create -f https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/v0.7.0/nvidia-device-plugin.yml

3. 查看节点 gpu 数量

1

kubectl get nodes "-o=custom-columns=NAME:.metadata.name,GPU:.status.allocatable.nvidia\.com/gpu"