矩池云里查看cuda版本
可以用下面的命令查看
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