1. AMD GPU
对于 AMD GPU,您可以使用 rocm-smi
工具,这是 AMD ROCm 软件包中的一部分。
安装 ROCm 工具包
首先,您需要安装 ROCm 工具包。以下是 Ubuntu 系统的安装步骤:
# 添加 ROCm 仓库
wget -qO - http://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add -
echo 'deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main' | sudo tee /etc/apt/sources.list.d/rocm.list
# 更新仓库并安装 ROCm
sudo