Kubernetes 配置节点支持 GPU
本文以 Debian 10, kubernetes 1.19.3 为例,进行配置 GPU 支持。
参考资料:
1. 前提条件
- 节点上已经安装了 nvidia 显卡驱动。驱动版本必须大于 384.81 。
- 节点上已经安装了
nvidia-docker 2.0
插件。安装过程见 docker 添加 nvidia 运行环境。 nvidia-container-runtime
作为默认的 runtime 。 即在 /etc/docker/daemon.json 中设置“default-runtime”: “nvidia”
2. 安装插件
kubectl create -f https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/v0.7.0/nvidia-device-plugin.yml
3. 查看节点 gpu 数量
kubectl get nodes "-o=custom-columns=NAME:.metadata.name,GPU:.status.allocatable.nvidia\.com/gpu"