使用java代码获取到GPU使用状况

要获取GPU使用状况，可以使用Java调用操作系统提供的API或者使用第三方库来实现。以下是获取GPU使用状况的两种方法：

方法一：使用Java调用操作系统提供的API

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class GPUInfo {
    public static void main(String[] args) {
        try {
            Process process = Runtime.getRuntime().exec("nvidia-smi");
            BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
            String line;
            while ((line = reader.readLine()) != null) {
                if (line.contains("Default")) {
                    System.out.println(line);
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

上述代码中，我们使用了Java的Runtime类来执行命令行命令，并使用BufferedReader类来读取命令行输出。在这个例子中，我们执行的命令是“nvidia-smi”，它是NVIDIA显卡驱动中提供的一个工具，用于查看GPU使用情况。我们通过读取命令行输出中包含“Default”的行来获取GPU使用情况。

方法二：使用第三方库

使用第三方库可以更方便地获取GPU使用情况。以下是使用JNvml库来获取GPU使用情况的示例代码：

import org.bytedeco.cuda.nvml.*;

public class GPUInfo {
    public static void main(String[] args) {
        Nvml.nvmlInit();
        NvmlDevice device = new NvmlDevice();
        Nvml.nvmlDeviceGetHandleByIndex(0, device);
        NvmlMemory memory = new NvmlMemory();
        Nvml.nvmlDeviceGetMemoryInfo(device, memory);
        System.out.println("Total memory: " + memory.total);
        System.out.println("Used memory: " + memory.used);
        NvmlUtilization utilization = new NvmlUtilization();
        Nvml.nvmlDeviceGetUtilizationRates(device, utilization);
        System.out.println("GPU utilization: " + utilization.gpu);
        Nvml.nvmlShutdown();
    }
}

上述代码中，我们使用了JNvml库来获取GPU的内存使用情况和GPU利用率。我们首先调用nvmlInit()方法初始化库，然后通过nvmlDeviceGetHandleByIndex()方法获取设备句柄，再调用nvmlDeviceGetMemoryInfo()方法获取内存使用情况，调用nvmlDeviceGetUtilizationRates()方法获取GPU利用率，最后调用nvmlShutdown()方法关闭库。

需要注意的是，使用第三方库需要先将库文件添加到Java项目中，并且需要按照库文件的使用说明进行配置和调用。