如何获得CUDA版本？

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

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

当前回答

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

其他回答

打开终端并运行以下命令：

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

您可以获得CUDA驱动程序版本、CUDA运行时版本的信息，以及GPU的详细信息。我的终端输出的图像示例如下。

您可以在此处找到图像。

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

import torch

print(torch.version.cuda)

在Ubuntu上：

Try

$cat/usr/local/cuda/version.txt或$cat/usr/local/cuda-8.0/version.txt

有时文件夹名为“Cuda版本”。

如果以上都不起作用，请尝试转到$/usr/local/找到Cuda文件夹的正确名称。

输出应类似于：CUDA版本8.0.61

使用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;
}

如果您在linux上运行：

dpkg -l | grep cuda

推荐文章