[发明专利]一种优化容器内显示进程信息的方法、系统、设备及介质

说明书

本发明公开了一种优化容器内显示进程信息的方法、系统、设备和存储介质，方法包括：设置能够劫持nvidia管理库函数的动态库文件；响应于接收到进程查询命令，获得运行在GPU上的所有进程，并获取进程在主机中的进程标识；根据进程标识在容器的目录中确定运行在容器中的相应进程；以及基于动态库文件劫持查询应用程序接口以输出在容器中的进程的信息。本发明通过设置动态库文件对查询应用程序接口进行劫持，并通过比对PID来确定容器中的相应进程，从而能够快速了解GPU设备和进程信息。

技术领域

本发明涉及容器领域，更具体地，特别是指一种优化容器内显示进程信息的方法、系统、计算机设备及可读介质。

背景技术

现今，深度学习和容器技术是开源社区和各大互联网公司争相采用的新兴技术，其中容器技术凭借着优秀的性能和相比于虚拟机本质上的优势，同时深度学习与容器结合使用也成为一种趋势和潮流，容器化运行深度学习训练或者推理业务日渐成熟并得到相当广泛应用和落地。Nvidia提供了诸多深度学习工具包，包括CUDA编程、GPU监控(nvidia-smi)等工具让开发人员能够最大利用GPU设备和对设备进行监控来优化实际应用。

一些常用GPU查询命令如nvidia-smi在HOST(主机)上可以正常使用并可以获取正在GPU设备上运行的进程信息。但是在容器中则不能完全发挥作用，nvidia-smi命令在容器内无法显示当前容器内在GPU设备正在进行计算的进程信息，这由于nvidia-smi是在驱动层面上的命令且不能感知容器内PID namespace，通过NVML(NVIDIA Management Library，nvidia管理库)接口可以看到该nvidia-smi返回的实际是HOST上该运行进程的PID，因此不能够通过该命令查到正在GPU上运行的进程。

发明内容

有鉴于此，本发明实施例的目的在于提出一种优化容器内显示进程信息的方法、系统、计算机设备及计算机可读存储介质，通过设置动态库文件对查询应用程序接口进行劫持，并通过比对PID来确定容器中的相应进程，从而能够快速了解GPU设备和进程信息。

基于上述目的，本发明实施例的一方面提供了一种优化容器内显示进程信息的方法，包括如下步骤：设置能够劫持nvidia管理库函数的动态库文件；响应于接收到进程查询命令，获得运行在GPU上的所有进程，并获取所述进程在主机中的进程标识；根据所述进程标识在容器的目录中确定运行在所述容器中的相应进程；以及基于所述动态库文件劫持查询应用程序接口以输出在所述容器中的所述进程的信息。

在一些实施方式中，所述设置能够劫持nvidia管理库函数的动态库文件包括：设置全局变量加载预先编译的动态库文件。

在一些实施方式中，所述根据所述进程标识在容器的目录中确定运行在所述容器中的相应进程包括：搜索所有容器内的进程，并将所述进程的第二进程标识写入所述容器的目录。

在一些实施方式中，所述根据所述进程标识在容器的目录中确定运行在所述容器中的相应进程包括：将所述进程标识与所述目录中的第二进程标识依次进行匹配。

在一些实施方式中，所述根据所述进程标识在容器的目录中确定运行在所述容器中的相应进程包括：响应于匹配成功，对匹配成功的所述第二进程标识对应的进程进行标识。

在一些实施方式中，所述基于所述动态库文件劫持查询应用程序接口以输出在所述容器中的所述进程的信息包括：将进行了标识的进程的信息进行输出。

在一些实施方式中，所述基于所述动态库文件劫持查询应用程序接口以输出在所述容器中的所述进程的信息包括：响应于检测到输出没有标识的进程的信息，基于所述动态库对所述查询应用程序接口进行劫持。