【注】:以下查看方式仅建立在已经安装完成的情况下!!!
关于Ubuntu的深度环境配置看参考之前的博文,link.
Windows下查看
1. 显卡型号
打开设备管理器——查看显示适配器。如下图,第一个是电脑带的集显,第二个是独立显卡 GTX 1660 SUPER
2. cuda版本
打开Nvidia控制面板,点击“帮助”——“系统信息”——“组件”,进行查看。
3. cudnn版本
进入该路径(cmd进也行,文件夹进也行)
[注]:路径中的v11.1是cuda的版本号,根据自己的cuda版本进行修改
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\include>
找到cudnn_version.h文件
可以直接选择记事本打开,翻到最后几行,看到下图所示就是cudnn的版本号。
Ubuntu下查看
这里先提一下有时候打开电脑忽然显示GPU不可用,像下图这样
处理方式(这样的前提是你的驱动之前是装好的才能这样处理):
step1:sudo apt-get install dkms
step2:sudo dkms install -m nvidia -v 410.79
1. 查看显卡信息
打开终端输入
$nvidia-smi
得到了以下的图,显示安装的驱动是455.28,cuda的版本是11.1
2. 查看cuda版本信息
打开终端输入
$nnvcc -V
得到了以下的图,显示cuda的版本是11.1,和上节中用nvidia-sim查询的结果相同。
3. 查看cudnn版本信息
直接用pytorch查看
得到下图