1. 程式人生 > 其它 >矩池雲裡檢視cuda版本

矩池雲裡檢視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

檢視文章

顯示卡,顯示卡驅動,nvcc, cuda driver,cudatoolkit,cudnn到底是什麼?