【查看显卡信息】如何查看nvidia-smi命令后的显卡信息

【查看显卡信息】如何查看nvidia-smi命令后的显卡信息

【先赞后看养成习惯】求点赞+关注+收藏！！！😊
一般我们用nvidia-smi进行查看当前电脑或者服务器的显卡信息，出来一大堆内容，很多小白根本看不懂，今天做一期内容来分析一下.
【windows、服务器如何查看显卡信息】

nvidia-smi

输入上述命令后，我们可以看到总共有7张显卡，每一张显卡都是NVIDIA TITAN X


(图片来源)
除了看显卡名称以及显存之外，这里有些参数如下图可以了解一下，其中我们最常需要看的参数就是当前占用显存的大小，像下面这种情况你要训练比较大的数据量的时候，需要选择第二块显卡（编号为1），只有这块显卡有足够的内存供你训练。

在代码中加入调用显卡命令：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

在终端调用显卡命令：

CUDA_VISIBLE_DEVICES=1 python train.py