java 获取GPU 信息（英伟达）

原创

已于 2023-08-28 14:15:27 修改 · 6.4k 阅读

16 ·

CC 4.0 BY-SA版权

文章标签：

#运维

于 2020-08-19 17:16:04 首次发布

本文介绍了一种通过命令行工具nvidia-smi获取GPU信息的方法，包括显存使用情况、GPU名称等，并提供了一个Java实现示例，同时推荐了英伟达的开源工具dcgm-exporter用于更详细的GPU监控。

方案一

获取已用显存，总显存，gpu 名称，使用率等等

命令获取gpu信息

public class ShellService {

    @Value("${micronaut.server.port}")
    private Integer myServerPort;

    @Inject
    private TrainModelRepository trainModelRepository;

    private static HashMap<String, Process> processHashMap = new HashMap<>();

    public String getGPU() throws IOException {
        Process process = null;
        try {
            if (Platform.isWindows()) {
                process = Runtime.getRuntime().exec("nvidia-smi.exe");
            } else if (Platform.isLinux()) {
                String[] shell = {"/bin/bash", "-c", "nvidia-smi"};
                process = Runtime.getRuntime().exec(shell);
            }

            process.getOutputStream().close();
        } catch (IOException e) {
            e.printStackTrace();
            throw new IndaiException("显卡不存在或获取显卡信息失败");
        }

        BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));

        StringBuffer stringBuffer = new StringBuffer();
        String line = "";
        while (null != (line = reader.readLine())) {
            stringBuffer.append(line + "\n");
        }

        return stringBuffer.toStr