linux下查看cpu个数，核数，线程数, GPU，内存

前言

有时我们也许想知道一个服务器上的性能如何。慢慢补充…

cpu

查看cpu个数

 cat /proc/cpuinfo | grep "physical id" | sort | uniq | wc -l

查看cpu核数

cat /proc/cpuinfo | grep "core id" | sort | uniq | wc -l

查看cpu线程数

cat /proc/cpuinfo | grep "processor" | sort | uniq | wc -l

GPU

会显示出型号，显存，利用率等。如果没有独显，没装驱动可能会显示找不到命令。

nvidia-smi

nvidia-smi -L

一秒刷新实时显示

watch -n 1 nvidia-smi

查看cuda版本

nvcc --version

python中 -V 经常等价于 --version

查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

但是如果不安装在该目录该命令就…没有用

使用pytorch查看cuda以及cudnn版本

import torch

print('device name:       ', torch.cuda.get_device_name())
print('pytorch version:   ', torch.__version__)
print('cuda version:      ', torch.version.cuda)
print('cudnn version:     ', torch.backends.cudnn.version())

内存

查看内存大小

按 MB单位查看

free -m

按GB单位查看

free -g

环境配置

判断是否配置好pytorch, cuda, cudnn有个简单的办法, 运行下面的文件即可。如果两个张量的输出都没问题，那么证明环境没问题。

import torch

print(torch.randn((2, 3)))
print(torch.randn((2, 3)).cuda())

系统相关

系统内核版本

uname -a

查看ubuntu版本

cat /proc/version

或查看linux版本

cat /etc/issue

查看当前python3进程

ps -ef|grep python3

查看那些进程占据了显存

fuser -v /dev/nvidia* 

有时候显卡被占了，可以用占用显卡的程序的PID号, 输入一下命令，大概知道实在哪里执行的，进而判断是谁用了

ll /proc/【PID号】