安装NVIDIA GPU支持
开启 GPU 支持 (云边共用)
在云端运行下面命令:
kubectl apply -f https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/v0.13.0/nvidia-device-plugin.yml
在边端修改edgecore.yaml
vim /etc/kubeedge/config/edgecore.yaml
# 修改以下部分
devicePluginEnabled: true
# 重启edgecore
systemctl restart edgecore.service
修改云边所有设备上的/etc/dockerdaemon.json
,添加如下内容:
{
"default-runtime": "nvidia",
"runtimes": {
"nvidia": {
"path": "nvidia-container-runtime",
"runtimeArgs": []
}
},
}
正常运行状态
在云端使用 kubectl get pods -A
查看nvidia相关pod,并使用 kubectl logs xxx -n xxx
或 kubectl describe pod xxx -n xxx
检查pod的情况。
云端pod情况:
边端pod情况:
在云端运行demo容器检查nvidia gpu是否能够正常获取:
kubectl run -i -t nvidia --image=jitteam/devicequery