矩池云里查看cuda版本

232 阅读1分钟

可以用下面的命令查看

cat /usr/local/cuda/version.txt

如果想用nvcc来查看可以用下面的命令

nvcc -V

如果环境内没有nvcc可以安装一下,教程是矩池云上如何安装nvcc

很多人会问为什么nvidia-smi里面的cuda会不一样,接下来我来介绍一下

可以看到环境内的CUDA 版本是10.2,而nvidia-smi的CUDA版本是10.0。很奇怪的是有时候绝大多数情况代码也能整成跑起来,stackoverflow上的一个解释如下:

CUDA有两个主要的API:runtime(运行时) API和driver API。这两个API都有对应的CUDA版本(如9.2和10.0等)。

用于支持driver API的必要文件(如libcuda.so)是由GPU driver installer安装的。nvidia-smi就属于这一类API。
用于支持runtime API的必要文件(如libcudart.so以及nvcc)是由CUDA Toolkit installer安装的。(CUDA Toolkit Installer有时可能会集成了GPU driver Installer)。nvcc是与CUDA Toolkit一起安装的CUDA compiler-driver tool,它只知道它自身构建时的CUDA runtime版本。它不知道安装了什么版本的GPU driver,甚至不知道是否安装了GPU driver。

综上,如果driver API和runtime API的CUDA版本不一致可能是因为你使用的是单独的GPU driver installer,而不是CUDA Toolkit installer里的GPU driver installer。

查cudnn版本

dpkg -l | grep cudnn

查看文章

显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn到底是什么?