如何获得CUDA版本？

是否有任何快速命令或脚本来检查安装的CUDA版本？

我在安装目录下找到了4.0的手册，但我不确定它是否是实际安装的版本。

当前回答

在使用CUDA 11.6.1的Windows 11上，这对我很有用：

cat "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\version.json"

其他回答

在我的cuda-11.60安装中，可以在/usr/local/cuda/version.json中找到信息。它包含完整的版本号（11.6.0而不是nvidia-smi显示的11.6）。

可按如下方式检索信息：

python -c 'import json; print(json.load(open("/usr/local/cuda/version.json"))["cuda"]["version"])'

[编辑的答案。感谢所有更正的人]

如果你跑步

nvidia-smi

您应该在命令输出的右上角找到安装的驱动程序支持的CUDA版本最高CUDA版本。至少我发现CUDA版本10.0的输出。，

使用CUDA运行时API C++包装器编程（注意：我是作者）：

auto v1 = cuda::version::maximum_supported_by_driver();
auto v2 = cuda::version::runtime();

这为您提供了一个cuda:：version_t结构，您可以对其进行比较，也可以打印/流式传输，例如：

if (v2 < cuda::version_t{ 8, 0 } ) {
    std::cerr << "CUDA version " << v2 << " is insufficient." std::endl;
}

在Windows 10上，我在“C:\Program Files\nvidia Corporation\NVSMI”中找到了nvidia-smi.exe；在cd之后插入该文件夹（在我的情况下不在PATH中）和'。\它显示了nvidia smi.exe

如果您安装了PyTorch，只需在IDE中运行以下代码：

import torch

print(torch.version.cuda)

推荐文章