获取 NVIDIA 设备 (/dev/nvidiaN) 访问数据的方法
要获取对 /dev/nvidiaN 的访问数据,您可以使用 nvidia-smi 命令或 NVIDIA Management Library (NVML) API。
使用 nvidia-smi 命令:
- 打开终端窗口。
- 输入以下命令获取特定 NVIDIA 设备(例如 /dev/nvidia0)的访问数据:
nvidia-smi --query-gpu=utilization.gpu,utilization.memory --format=csv --id=N
请将 N 替换为您想要获取访问数据的设备 ID(例如 0)。
使用 NVML API:
- 编写一个程序来使用 NVML API 获取对 /dev/nvidiaN 的访问数据。您可以使用 C、C++、Python 等编程语言。
- 在程序中,使用 nvmlDeviceGetUtilizationRates 函数获取 GPU 的利用率数据。示例代码如下(使用 Python):
import pynvml
pynvml.nvmlInit()
# 获取设备数量
device_count = pynvml.nvmlDeviceGetCount()
# 选择要获取访问数据的设备
device_index = N # 将 N 替换为您想要获取访问数据的设备索引
# 打开设备
handle = pynvml.nvmlDeviceGetHandleByIndex(device_index)
# 获取 GPU 利用率数据
utilization = pynvml.nvmlDeviceGetUtilizationRates(handle)
print('GPU 利用率: {}%'.format(utilization.gpu))
print('内存利用率: {}%'.format(utilization.memory))
pynvml.nvmlShutdown()
请将 N 替换为您想要获取访问数据的设备索引(例如 0)。
请注意,要使用 NVML API 获取访问数据,您需要先安装 NVIDIA 驱动和 NVML 库。
原文地址: https://www.cveoy.top/t/topic/qDva 著作权归作者所有。请勿转载和采集!