获取CPU使用率(golang)

语言:golang
适用系统:linux
用途:获取CPU使用率
原文地址:https://stackoverflow.com/questions/11356330/how-to-get-cpu-usage

package main

import (
    "fmt"
    "io/ioutil"
    "strconv"
    "strings"
    "time"
)

func getCPUSample() (idle, total uint64) {
    contents, err := ioutil.ReadFile("/proc/stat")
    if err != nil {
        return
    }
    lines := strings.Split(string(contents), "\n")
    for _, line := range(lines) {
        fields := strings.Fields(line)
        if fields[0] == "cpu" {
            numFields := len(fields)
            for i := 1; i < numFields; i++ {
                val, err := strconv.ParseUint(fields[i], 10, 64)
                if err != nil {
                    fmt.Println("Error: ", i, fields[i], err)
                }
                total += val // tally up all the numbers to get total ticks
                if i == 4 {  // idle is the 5th field in the cpu line
                    idle = val
                }
            }
            return
        }
    }
    return
}

func main() {
    idle0, total0 := getCPUSample()
    time.Sleep(3 * time.Second)
    idle1, total1 := getCPUSample()

    idleTicks := float64(idle1 - idle0)
    totalTicks := float64(total1 - total0)
    cpuUsage := 100 * (totalTicks - idleTicks) / totalTicks

    fmt.Printf("CPU usage is %f%% [busy: %f, total: %f]\n", cpuUsage, totalTicks-idleTicks, totalTicks)
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Golang 中,可以通过使用 `github.com/shirou/gopsutil/cpu` 和 `github.com/mindprince/gonvml` 等第三方库获取 Linux 系统的 CPU 和 GPU 使用率。下面是获取 CPU 和 GPU 使用率的示例代码: ```go 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 系统上安装。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值