天天看點

cuda nvcc版本不一緻_CUDA版本檢測

當CUDA9安裝完成後,需要進行檢測:

1) cat /usr/local/cuda/version.txt #這種方法不太準,得到9.0.176

2) nvcc --version #這種方法準。如果nvcc沒有安裝,則sudo apt install nvidia-cuda-toolkit。檢查出來是7.5.17

3)通過代碼來查:

cudaDriverGetVersion(&driver_version); #擷取cuda版本。得到9.0

cudaRuntimeGetVersion(&runtime_version); #擷取cuda運作時版本。得到7.5

這種方法,例子是https://blog.csdn.net/langb2014/article/details/51348501

還有一個中文的例子https://www.2cto.com/kf/201708/666635.html

其中,CMakeLists.txt的寫法,例子是https://blog.csdn.net/jz4626828/article/details/79878945

4)檢測顯示卡型号及NVIDIA驅動版本

檢視GPU型号 lspci | grep -i nvidia 得到M4000

檢視NVIDIA驅動版本 sudo dpkg --list | grep nvidia-* 得到384.130

根據不同版本cuda對應的NVIDIA驅動版本(不同版本cuda對應的NVIDIA驅動版本 - LearnFromNow - 部落格園),這個驅動應該對應cuda9.0

5)進一步查是否環境變量的問題,參見 Ubuntu16.04下安裝多版本cuda和cudnn - tunhuzhuang1836的部落格 - CSDN部落格

在~/.bashrc中檢視有無類似的語句,如有改成下面的樣子,沒有則增加:

export PATH="$PATH:/usr/local/cuda/bin"

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

export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda/lib64"

然後繼續nvcc --version還是得到7.5。

6)檢查nvcc

which nvcc,發現是/usr/bin/nvcc,而實際上nvcc是應該在/usr/local/cuda/bin/nvcc。

是以,sudo gedit /usr/bin/nvcc,把裡面的内容"exec /usr/lib/nvidia-cuda-toolkit/bin/nvcc" 改成"exec /usr/local/cuda/bin/nvcc"

然後繼續nvcc --version這次得到9.0

繼續閱讀