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

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


当前回答

我们有三种方法来检查版本:在我的案例中,以下是输出:-方式1:-

cat /usr/local/cuda/version.txt

输出:-

CUDA Version 10.1.243

途径2:-

nvcc --version

输出:-

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

途径3:-

/usr/local/cuda/bin/nvcc --version

输出:-

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

途径4:-

nvidia-smi
NVIDIA-SMI 450.36.06    Driver Version: 450.36.06    CUDA Version: 11.0

输出不相同。不知道为什么会这样。

其他回答

对于CUDA版本:

nvcc --version

或者使用,

nvidia-smi

对于cuDNN版本:

对于Linux:

使用以下方法查找cuDNN的路径:

$ whereis cuda
cuda: /usr/local/cuda

然后使用该命令从头文件获取版本,

$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

对于Windows,

使用以下方法查找cuDNN的路径:

C:\>where cudnn*
C:\Program Files\cuDNN7\cuda\bin\cudnn64_7.dll

然后使用它从头文件转储版本,

type "%PROGRAMFILES%\cuDNN7\cuda\include\cudnn.h" | findstr CUDNN_MAJOR

如果您在Windows上获得CUDA的两个不同版本-nvcc和NVIDIA smi显示的不同CUDA版本

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

在Ubuntu上:

Try

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

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

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

输出应类似于:CUDA版本8.0.61

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

import torch

print(torch.version.cuda)

如果安装了多个CUDA,则系统中加载的CUDA是与“nvcc”关联的CUDA。因此,“nvcc--version”显示了您想要的内容。