Linux深度学习各种查看操作

查看GPU使用情况

nvidia-smi

查看 CUDA 版本

cat /usr/local/cuda/version.txt
或
cat /usr/local/cuda/version.json

nvidia-smi

nvcc -V

nvcc --version

这里有两点需要注意：

（1）nvidia-smi显示的是兼容的CUDA最高版本，nvcc --version显示的是运行CUDA

（2）nvcc --version和nvcc -V都是用于查看NVIDIA CUDA编译器（nvcc）的版本信息的命令。两个命令的区别在于输出的详细程度。如果你需要获取更多关于CUDA编译器的版本信息，可以使用nvcc --version命令；如果只需要获取简单的版本号信息，可以使用nvcc -V命令

查看显卡型号

 lspci | grep -i vga

将得到的密匙进入网址http://pci-ids.ucw.cz/mods/PC/10de?action=help?help=pci输入即可得知显卡型号

检查驱动是否安装成功

nvidia-smi

nvidia-settings

检查CUDNN版本

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

检查torch GPU是否可以使用

进入python环境

python

检查torch调用情况，若返回为True，则使用的是GPU版本的torch，若为False，则为CPU版本

import  torch 
torch.cuda.is_available()

查看GPU型号

print(torch.cuda.get_device_name(0))

查看cuda设备数量

print(torch.cuda.device_count())

查看当前使用的cuda编号

print(torch.cuda.current_device())

查看设备容量

print(torch.cuda.get_device_capability(0))

查看torch版本

print(torch.__version__)