Linux中GPU相关命令

皮肤小白生

已于 2024-03-14 15:38:03 修改

阅读量563

点赞数 7

分类专栏： Linux 文章标签： linux 运维服务器

于 2024-03-14 10:54:46 首次发布

本文链接：https://blog.csdn.net/weixin_46587777/article/details/136701589

版权

Linux 专栏收录该内容

13 篇文章 1 订阅

订阅专栏

本文介绍了如何在Linux系统中查看显卡信息，包括使用`lspci`和`nvidia-smi`命令，以及如何通过`CUDA_VISIBLE_DEVICES`环境变量指定GPU使用。还提及了NvidiaPrime工具用于切换显卡和其操作注意事项。

摘要由CSDN通过智能技术生成

Linux查看显卡信息：

lspci | grep -i vga

使用nvidia GPU可以：

lspci | grep -i nvidia

前边的序号 "00:0f.0"是显卡的代号(这里是用的虚拟机);

查看指定显卡的详细信息用以下指令：

lspci -v -s 00:0f.0

Linux查看Nvidia显卡信息及使用情况

Nvidia自带一个命令行工具可以查看显存的使用情况：

nvidia-smi

表头释义：

Fan：显示风扇转速，数值在0到100%之间，是计算机的期望转速，如果计算机不是通过风扇冷却或者风扇坏了，显示出来就是N/A；
Temp：显卡内部的温度，单位是摄氏度；
Perf：表征性能状态，从P0到P12，P0表示最大性能，P12表示状态最小性能；
Pwr：能耗表示；
Bus-Id：涉及GPU总线的相关信息；
Disp.A：是Display Active的意思，表示GPU的显示是否初始化；
Memory Usage：显存的使用率；
Volatile GPU-Util：浮动的GPU利用率；
Compute M：计算模式；
下边的Processes显示每块GPU上每个进程所使用的显存情况。

如果要周期性的输出显卡的使用情况，可以用watch指令实现：

watch -n 10 nvidia-smi

命令行参数-n后边跟的是执行命令的周期，以s为单位。

第二个命令：nvidia-smi -L

该命令用于列出所有可用的 NVIDIA 设备信息。

指定空闲的GPU运行python程序

CUDA_VISIBLE_DEVICES=0,1 python test.py

📘 在python程序中指定GPU（通常使用该设置）

在 train.py 最上方，设定使用的GPU 编号， 当这两行代码注释掉时，训练会自动使用该服务器的所有资源
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

NVIDIA提供了一个切换显卡的命令：
apt install nvidia-prime
sudo prime-select nvidia # 切换nvidia显卡
sudo prime-select intel # 切换intel显卡
sudo prime-select query # 查看当前使用的显卡

注意：每一次切换显卡都需要重新启动电脑才能生效。