(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202211066506.3 (22)申请日 2022.09.01 (71)申请人 深圳依时货拉 拉科技有限公司 地址 518000 广东省深圳市福田区梅林街 道梅丰社区梅华路105号多丽工业区 科技楼3层3 07 (72)发明人 张浩　黄政　赫子敬　苏海红　 王海艳　 (74)专利代理 机构 深圳瑞天谨诚知识产权代理 有限公司 4 4340 专利代理师 温青玲 (51)Int.Cl. G06F 9/48(2006.01) G06F 9/50(2006.01) G06F 9/455(2006.01) (54)发明名称 一种Notebook GPU任务调度方法、 装置、 计 算机设备及计算机可读存 储介质 (57)摘要 本发明提供了一种Notebook  GPU任务调度 方法， 包括： 创建GPU节点并且注册GPU实例到 Notebook管理平台GPU实例列表； 在Notebook管 理平台上启动Notebook  GPU任务并运行； 根据 Notebook  GPU任务运行情况进行Notebook  GPU 任务回收。 本发明通过采用Kubernetes调度 Notebook  GPU任务到指定的GPU节点， 当用户不 再需要使用GP U节点时可手动停 止Notebook服务 和停止GPU节点， 或者是当达到GP U任务回收条件 时自动停止停止Notebook服务和停止GPU节点， 从而更加高效的使用GPU资源以降低企业使用成 本。 权利要求书2页 说明书4页 附图1页 CN 115408131 A 2022.11.29 CN 115408131 A 1.一种Notebook GPU任务调度方法， 其特 征在于， 所述方法包括： S101、 创建 GPU节点并且注 册GPU实例到N otebook管理平台GPU实例列表； S102、 在所述 Notebook管理平台上启动N otebook GPU任务并运行； S103、 根据所述 Notebook GPU任务运行情况进行N otebook GPU任务回收。 2.如权利要求1所述的方法， 其特征在于， 所述创建GPU节点并且注册GPU实例到 Notebook管理平台的GPU实例列表， 包括： 申请GPU资源； 初始化GPU节点； 将所述GPU节点纳管到Kubernetes集群中； 和 注册GPU实例到N otebook管理平台GPU实例列表。 3.如权利 要求2所述的方法， 其特征在于， 所述Notebook管理平台在数据库中保存所述 GPU实例的相关信息， 所述GPU实例的相关信息包括GPU型号、 卡数、 GPU机器CPU、 内存、 开关 机状态、 或是否绑定N otebook服务中的一个或多个。 4.如权利要求1所述的方法， 其特征在于， 在所述Notebook管理平台上启动Notebook   GPU任务并运行， 包括： 创建Notebook实例； 判断是否有GPU实例与所述Notebook实例匹配， 当没有GPU实例匹配时通知用户不能创 建该资源请求的GPU任务， 当有GPU实例匹配时启动GPU节 点并且将该GPU实例任务状态标记 为运行中； 将所述GPU节点与所述 Notebook实例进行绑定； 调用Kubernetes API Server部署Notebook服务到Kub ernetes集群上匹配到的GPU机 器实例上， 并且在N otebook中运行GPU任务。 5.如权利要求4所述的方法， 其特征在于， 所述判断是否有GPU实例与所述Notebook实 例匹配， 具体是： 所述Notebook管理平台对用户选择GPU相关资源的请求进行判断， 通过查询Notebook 管理平台GPU实例列表查看是否有满足条件的GPU实例。 6.如权利要求1所述的方法， 其特征在于， 所述根据Notebook  GPU任务运行情况进行 Notebook GPU任务回收， 具体是： 当所述Notebook  GPU任务空闲时触发Notebook  GPU任务自动回收， Notebook服务调用 所述Notebook管理平台， 所述Notebook管理平台接收到Notebook自动回收操作后停掉 Notebook服务， 之后停止GPU机器， 并解绑GPU实例列表中所述GPU与N otebook的绑定关系； 或者是： 当用户不再使用所述Notebook  GPU任务， 在所述Notebook管理平台上主动停止 Notebook服务， 所述Notebook管理平台停止掉Notebook服务后针对GPU任务去停止GPU机 器， 并解绑GPU实例列表中所述GPU与N otebook的绑定关系。 7.如权利要求6所述的方法， 其特 征在于， 所述 Notebook GPU任务空 闲具体是： 当所述Notebook管理平台的前端和后端之间没有任务交互的时长超过GPU任务自动回 收时长阈值， 所述GPU任务自动回收时长阈值是根据任务设定的。 8.一种Notebook GPU任务调度装置， 其特 征在于， 所述装置包括：权　利　要　求　书 1/2 页 2 CN 115408131 A 2节点增加模块， 用于创建GPU节点并且注册GPU实例到Notebook管理平台的GPU实例列 表； 任务部署模块， 用于在所述 Notebook管理平台上启动N otebook GPU任务并运行； 任务回收模块， 用于根据所述Notebook  GPU任务运行情况进行Notebook  GPU任务回 收。 9.一种计算机设备， 包括： 一个或多个处 理器； 存储器； 以及 一个或多个计算机程序， 所述处理器和所述存储器通过总线连接， 其中所述一个或多 个计算机程序被存储在所述存储器中， 并且被配置成由所述一个或多个处理器执行， 其特 征在于， 所述处理器执行所述计算机程序时实现如权利要求1至7任一项所述的Notebook   GPU任务调度方法的步骤。 10.一种计算机可读存储介质， 所述计算机可读存储介质存储有计算机程序， 其特征在 于， 所述计算机程序被处理器执行时实现如权利要求1至7任一项 所述的Notebook  GPU任务 调度方法的步骤。权　利　要　求　书 2/2 页 3 CN 115408131 A 3