1 CUDA Driver VersionCUDA 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,如下图所示

如何获取CUDA版本号?CUDA Driver Version和CUDA Runtime Version的区别是什么?-StubbornHuang Blog

1.2 错误的方法

1.2.1 使用nvidia-smi获取

我们也经常使用nvidia-smi获取显卡运行信息,可是这个命令显示的CUDA版本号实际上是CUDA Driver Version,也就是本机驱动所支持的最大CUDA版本,而不是你实际使用的CUDA Runtime Version。

如何获取CUDA版本号?CUDA Driver Version和CUDA Runtime Version的区别是什么?-StubbornHuang Blog

1.2.2 在Nvidia控制面板查看CUDA版本号

在Nvidia控制面板的系统信息所查看CUDA版本号也是CUDA Driver Version,而不是你实际使用的CUDA Runtime Version。

如何获取CUDA版本号?CUDA Driver Version和CUDA Runtime Version的区别是什么?-StubbornHuang Blog