矩池雲裡檢視cuda版本
阿新 • • 發佈:2021-10-11
可以用下面的命令檢視
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