查看显卡型号:
lspci |grep VGA
(lspci是linux查看硬件信息的命令),屏幕会打印出主机的集显几独显信息
python中查看显卡型号
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
查看nvidia芯片信息:
lspci |grep -i nvidia
会打印出nvidia系列的硬件信息,如果主机安装了没有视频输出的GPU(如tesla系列),这个命令会很有用
查看 CUDA 版本
cat /usr/local/cuda/version.txt
查看 cudnn 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
查看Navicat GPU版本
nvidia-smi
//10s显示一次
watch -n 10 nvidia-smi
nvidia-smi -l
Python 脚本来查看 CUDNN 版本
import os
os.system('cat /usr/local/cuda/version.txt')
#os.system('whereis cudnn')
try:
head_file = open('/usr/local/cuda/include/cudnn.h')
except:
head_file = open('/usr/include/cudnn.h')
lines = head_file.readlines()
for line in lines:
line = line.strip()
if line.startswith('#define CUDNN_MAJOR'):
line = line.split('#define CUDNN_MAJOR')
n1 = int(line[1])
continue
if line.startswith('#define CUDNN_MINOR'):
line = line.split('#define CUDNN_MINOR')
n2 = int(line[1])
continue
if line.startswith('#define CUDNN_PATCHLEVEL'):
line = line.split('#define CUDNN_PATCHLEVEL')
n3 = int(line[1])
break
print('CUDNN Version ', str(n1)+'.'+str(n2)+'.'+str(n3))
Windows 查看 CUDA 版本
方法 1:
进入以下目录
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
即可安装的 CUDA 版本
方法 2:
打开 cmd,输入
nvcc --version