golang 获取linux cpu gpu 使用率

在 Golang 中，可以通过使用 `github.com/shirou/gopsutil/cpu` 和 `github.com/mindprince/gonvml` 等第三方库获取 Linux 系统的 CPU 和 GPU 使用率。下面是获取 CPU 和 GPU 使用率的示例代码：

package main

import (
    "fmt"
    "time"

    "github.com/mindprince/gonvml"
    "github.com/shirou/gopsutil/cpu"
)

func main() {
    // 获取 CPU 使用率
    percent, err := cpu.Percent(time.Second, false)
    if err != nil {
        fmt.Printf("Error: %s\n", err)
    }
    fmt.Printf("CPU Usage: %v%%\n", percent[0])

    // 获取 GPU 使用率
    err = gonvml.Initialize()
    if err != nil {
        fmt.Printf("Error: %s\n", err)
    }
    defer gonvml.Shutdown()

    count, err := gonvml.DeviceCount()
    if err != nil {
        fmt.Printf("Error: %s\n", err)
    }
    for i := 0; i < count; i++ {
        handle, err := gonvml.DeviceHandleByIndex(uint(i))
        if err != nil {
            fmt.Printf("Error: %s\n", err)
        }

        utilization, err := handle.UtilizationRates()
        if err != nil {
            fmt.Printf("Error: %s\n", err)
        }
        fmt.Printf("GPU %d Usage: GPU %v%%, Memory %v%%\n", i, utilization.GPU, utilization.Memory)
    }
}

在这个示例代码中，我们使用了 `github.com/shirou/gopsutil/cpu` 和 `github.com/mindprince/gonvml` 两个第三方库来获取 CPU 和 GPU 使用率。

对于 CPU 使用率，我们使用了 `cpu.Percent` 函数来获取当前 CPU 使用率。该函数的第一个参数为时间间隔，第二个参数为是否包含全部 CPU 核心。返回值为一个 `[]float64` 类型的切片，其中每个元素表示一个 CPU 核心的使用率。

对于 GPU 使用率，我们首先通过 `gonvml.Initialize` 函数初始化 GPU 相关信息。然后我们获取 GPU 数量，并通过 `gonvml.DeviceHandleByIndex` 函数获取每个 GPU 的句柄。最后，我们使用 `handle.UtilizationRates` 函数来获取 GPU 和内存的使用率。

注意，为了能够获取 GPU 使用率，需要先安装 NVIDIA 显卡驱动和 CUDA 工具包，并安装 `gonvml` 库。可以通过 `sudo apt-get install nvidia-cuda-toolkit` 命令在 Ubuntu 系统上安装。