自己diy一个smi（可以监控gpu,cpu,memory）

diy-smi 1.0（DIY System management interface)

在我们训练自己深度学习模型时，我们想看GPU信息，也想看CPU和内存信息。我常用的是nvidia-smi和htop，同时要开两个窗口，并且很多参数，我其实并不想查看。我们可以根据我们想要查看硬件那些信息，来自己diy一个监控窗口。

当然，你也可以在我的基础上，继续添加自己想要的元素，制作更炫酷的smi。

使用：

0️⃣ 先将代码下载，克隆，复制到本地。Github diy-smi 1.0

1️⃣ 首先需要安装三个包，分别是用于收集gpu信息，cpu信息，美化我们的输出窗口。

pip install nvidia-ml-py3 -i  https://pypi.mirrors.ustc.edu.cn/simple
pip install psutil -i  https://pypi.mirrors.ustc.edu.cn/simple
pip install rich -i  https://pypi.mirrors.ustc.edu.cn/simple

2️⃣ 然后运行我们的脚本就可以了

python diy-smi.py

3️⃣ 如果想用 watch -n 1命令，输出可能会没有颜色，在我电脑上没有颜色。

watch -c -n 1 python diy-smi.py --color

效果展示：

Ubantu terminal:

Windows Terminal Preview: