是否有任何快速命令或脚本来检查安装的CUDA版本?
我在安装目录下找到了4.0的手册,但我不确定它是否是实际安装的版本。
是否有任何快速命令或脚本来检查安装的CUDA版本?
我在安装目录下找到了4.0的手册,但我不确定它是否是实际安装的版本。
当前回答
通过在终端中键入以下内容,可以获得cuda版本:
$ nvcc -V
# below is the result
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
或者,可以先使用以下方法查找安装目录,手动检查版本:
$ whereis -b cuda
cuda: /usr/local/cuda
然后cd到该目录并检查CUDA版本。
其他回答
如果您已安装CUDA SDK,则可以运行“deviceQuery”查看CUDA的版本
如果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
如果安装了多个CUDA,则系统中加载的CUDA是与“nvcc”关联的CUDA。因此,“nvcc--version”显示了您想要的内容。
在以下时间之后找到我的:
whereis cuda
at
cuda:/usr/lib/cuda/usr/include/cuda.h
with
nvcc --version
CUDA版本9.1.85