Ubuntu下查看cuda版本的两种方法

大家好，又见面了，我是你们的朋友全栈君。

参考资料：https://blog.csdn.net/qq_16525279/article/details/80662217

  在安装Pytorch等深度学习框架的时候，我们往往需要检查是否和cuda版本匹配。在Ubuntu系统下查看cuda版本的方法，我发现有两个。

方法一：

  比较常用的一个方法是使用如下命令：

cat /usr/local/cuda/version.txt

使用该命令的效果如下图所示：

方法一使用效果

方法二：

  方法一主要是依据cuda安装时保存的关于版本的txt文件。但是这个txt文件有可能被清空或者是误删，再用方法一查询的结果就为空。这个时候可以用第二种方法，使用如下命令：

nvidia-smi

这个命令原本是用来查看显存占用情况的，但是也会附带显示cuda的版本信息，显示的效果如下图所示：

方法二使用效果

附：对于有些网友反馈说，两个方法显示结果不一致的情况，一开始我以为是cuda后续有更新导致的，不过有一篇博文解释了这个情况，据说是有驱动（driver）API版本和运行（runtime）API的区别，这位作者建议参考方法一的结果安装tensorflow和pytorch等程序，大家可以参考一下这篇博文。

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/131557.html原文链接：https://javaforall.cn