是否有任何快速命令或脚本来检查安装的CUDA版本?
我在安装目录下找到了4.0的手册,但我不确定它是否是实际安装的版本。
是否有任何快速命令或脚本来检查安装的CUDA版本?
我在安装目录下找到了4.0的手册,但我不确定它是否是实际安装的版本。
当前回答
正如Jared在命令行的评论中提到的:
nvcc --version
(或/usr/local/cuda/bin/nvcc-version)提供cuda编译器版本(与工具包版本匹配)。
从应用程序代码中,可以使用
cudaRuntimeGetVersion()
或驱动程序API版本
cudaDriverGetVersion()
正如Daniel所指出的那样,deviceQuery是一个SDK示例应用程序,可以查询上面的内容以及设备功能。
正如其他人所指出的,您也可以使用(例如,在Mac或Linux上)检查version.txt的内容
cat /usr/local/cuda/version.txt
但是,如果安装了CUDA工具包的另一个版本,而不是从/usr/local/CUDA符号链接的版本,则如果PATH中的另一版本早于上述版本,则可能会报告版本不准确,因此请谨慎使用。
其他回答
对于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版本
使用tensorflow:
import tensorflow as tf
from tensorflow.python.platform import build_info as build
print(f"tensorflow version: {tf.__version__}")
print(f"Cuda Version: {build.build_info['cuda_version']}")
print(f"Cudnn version: {build.build_info['cudnn_version']}")
tensorflow版本:2.4.0
Cuda版本:11.0
Cudnn版本:8
在Ubuntu Cuda V8上:
$ cat /usr/local/cuda/version.txt
您还可以了解CUDA版本的安装情况:
$ ls -l /usr/local | grep cuda
这会给你这样的东西:
lrwxrwxrwx 1 root root 9 Mar 5 2020 cuda -> cuda-10.2
drwxr-xr-x 16 root root 4096 Mar 5 2020 cuda-10.2
drwxr-xr-x 16 root root 4096 Mar 5 2020 cuda-8.0.61
给定一个正常的PATH,cuda指向的版本应该是活动版本(本例中为10.2)。
注意:只有当您愿意假设CUDA安装在/usr/local/CUDA下时,这才有效(这对于具有默认位置的独立安装程序是正确的,但对于CUDA集成为软件包的发行版则不正确)。参考:@einpoklum的评论。
在Windows 10上,我在“C:\Program Files\nvidia Corporation\NVSMI”中找到了nvidia-smi.exe;在cd之后插入该文件夹(在我的情况下不在PATH中)和'。\它显示了nvidia smi.exe
你可能会发现CUDA-Z很有用,这里是他们网站上的一句话:
“这个程序诞生于另一个Z-实用程序的模仿,例如CPU-Z和GPU-Z。CUDA-Z显示了CUDA支持的GPU和GPGPU的一些基本信息。它与nVIDIA Geforce、Quadro和Tesla卡以及ION芯片组一起工作。”
http://cuda-z.sourceforge.net/
在支持选项卡上,有源代码的URL:http://sourceforge.net/p/cuda-z/code/并且下载实际上不是安装程序,而是可执行文件本身(没有安装,所以这是“快速”的)。
此实用程序提供了大量信息,如果您需要了解它是如何派生的,可以查看源代码。您可以搜索与此类似的其他实用程序。