大家好,又见面了,我是你们的朋友全栈君。
参考资料: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等程序,大家可以参考一下这篇博文。
发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/131557.html原文链接:https://javaforall.cn
【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛
【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...