【查看显卡信息】如何查看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