记录一些常用操作

使用清华源安装库

pip install pygame -i https://pypi.tuna.tsinghua.edu.cn/simple

要实时监控 GPU 使用率(GPU Utilization) 和 显存占用(GPU Memory Usage),可以使用以下方法,按 每秒刷新 查看数据:

watch -n 1 nvidia-smi

会出现

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 535.54.03    Driver Version: 535.54.03    CUDA Version: 12.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  NVIDIA RTX 4090     On   | 00000000:01:00.0  On |                  Off |
| 30%   45C    P2    100W / 450W|   10240MiB / 24564MiB |     98%      Default |
+-------------------------------+----------------------+----------------------+

或者使用

watch -n 1 nvidia-smi --query-gpu=utilization.gpu,memory.used --format=csv

出现

utilization.gpu [%], memory.used [MiB]
98 %, 10240 MiB

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐