如何获得CUDA版本？

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

我在安装目录下找到了4.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;
}

其他回答

在以下时间之后找到我的：

whereis cuda

cuda:/usr/lib/cuda/usr/include/cuda.h

with

nvcc --version

CUDA版本9.1.85

如果nvcc--版本不适合您，请使用cat/usr/local/cuda/version.txt

如果您已安装CUDA SDK，则可以运行“deviceQuery”查看CUDA的版本

在我的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"])'

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

import torch

print(torch.version.cuda)

推荐文章