如何获取CUDA版本?

908

有没有快速的命令或脚本来检查安装的CUDA版本?

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


4
请参阅:如何验证CuDNN安装? - Martin Thoma
7
这个问题的目标操作系统是哪个? - nbro
17
@JaredHoberock nvcc --version 命令会提示 The program 'nvcc' is currently not installed. You can install it by typing: sudo apt install nvidia-cuda-toolkit,但是 nvidia-smi 命令显示 CUDA 版本为 10.1。 - mrgloom
4
但运行 cat /usr/local/cuda/version.txt 命令可以得到更精确的版本信息:CUDA 版本 10.1.168 - mrgloom
如果您正在寻找实际版本数字的字符串,在Unix中,您可以使用nvidia-smi混合获取版本,使用grep识别行,并使用sed删除不必要的字符:nvidia-smi | grep -o 'CUDA Version: [0-9].\.[0-9]' | sed 's/.*: //' - Eduardo Pignatelli
显示剩余2条评论
31个回答

-3

如果nvcc和nvidia-smi之间存在版本不匹配,则会使用不同版本的cuda作为驱动程序和运行时环境。

为确保使用相同版本的CUDA驱动程序,您需要将CUDA添加到系统路径中。

首先运行whereis cuda并找到cuda驱动程序的位置。

然后进入.bashrc文件并修改路径变量,使用变量“LD_LIBRARY_PATH”设置搜索目录的优先顺序。

例如

$ whereis cuda
cuda: /usr/lib/cuda /usr/include/cuda.h /usr/local/cuda

CUDA已安装在/usr/local/cuda目录下,现在我们需要编辑.bashrc文件,并添加路径变量:
vim  ~/.bashrc
export PATH="/usr/local/cuda/bin:${PATH}"

在此行后,将目录搜索路径设置为:

export LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}"

然后保存 .bashrc 文件。并刷新它:

$ source ~/.bashrc

这将确保您拥有nvcc -V和nvidia-smi以使用相同版本的驱动程序。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接