如何获取CUDA版本号?CUDA Driver Version和CUDA Runtime Version的区别是什么?
1 CUDA Driver Version和CUDA Runtime Version的区别
CUDA Driver Version是跟Nvidia的GPU驱动(nvidia-driver)绑定在一起的,指的是当前GPU驱动所能驱动的CUDA的最高版本。
CUDA Runtime Version是在本机上实际安装的CUDA版本,是Pytorch或者其他深度学习框架实际调用的CUDA版本,这个版本应该低于上面所述的CUDA Driver Version。
1.1 如何正确获取CUDA Runtime Version
一般来讲,我们在电脑上安装CUDA成功之后所需获取的都是CUDA Runtime Version,而不是CUDA Driver Version,我们更加关心我们训练模型时最终使用的是哪个版本。
所以正确的获取CUDA Runtime Version的方法是通过
nvcc -V
或者
nvcc --version
获取CUDA Runtime Version,如下图所示
1.2 错误的方法
1.2.1 使用nvidia-smi获取
我们也经常使用nvidia-smi
获取显卡运行信息,可是这个命令显示的CUDA版本号实际上是CUDA Driver Version,也就是本机驱动所支持的最大CUDA版本,而不是你实际使用的CUDA Runtime Version。
1.2.2 在Nvidia控制面板查看CUDA版本号
在Nvidia控制面板的系统信息所查看CUDA版本号也是CUDA Driver Version,而不是你实际使用的CUDA Runtime Version。
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:如何获取CUDA版本号?CUDA Driver Version和CUDA Runtime Version的区别是什么?
原文链接:https://www.stubbornhuang.com/2714/
发布于:2023年07月03日 17:13:49
修改于:2023年07月03日 17:13:49
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。
评论
50