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

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


当前回答

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

其他回答

如果您已安装CUDA SDK,则可以运行“deviceQuery”查看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

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

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

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

您可以在此处找到图像。

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

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