CUDA、cudnn安裝教程

目錄

• CUDA安裝
  • 歷史版本CUDA安裝
  • 安裝小測
• CUDNN安裝
  • 選擇系統對應的版本(我的環境切換為windows10了)
• cudnn安裝版本查詢

CUDA安裝

資源鏈接：
歷史版本的安裝
當前最新的CUDA安裝
當前以ubuntu18.04的服務器環境，安裝cuda10.02

歷史版本CUDA安裝

選擇好對應版本之后，選擇操作系統等一系列信息，你可以看見如下信息：

安裝命令：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

如圖所示，根據命令在命令行執行就可以了

安裝小測

在終端執行下面的命令進行cuda測試：

nvidia-smi

CUDNN安裝

查找歷史版本及其與CUDA的對應關系

選擇系統對應的版本(我的環境切換為windows10了)

1. 第一步
   
2. 第二步
   
3. 第三步
   
4. 第四步
   
5. 下載(windows10對應版本)
   cuda10.2對應的cudnn7.6.5.23:

https://developer.download.nvidia.cn/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.2_20191118/cudnn-10.2-windows10-x64-v7.6.5.32.zip?lSdULl8SFl_ABsp7SYDWfquUYycvAs1BC5IKjQckJJcf7RuyDq7lK1anjgmsAvgtkibFHNgBCdf-W4-Q0g7lleaLBTywu4xHrSAXBfrMaghufQFFdYhrWk6Cf6SQo8zJa0ydTBX91oAMXQnjIIhQGlhjoAkuTPDlPn3TkZv3A3q_Y29i1yopf4nIOe9TPUrXou2FpqQqcdV8EVcHU5XUxh9xl7CF64yDECsorkI

cudnn安裝版本查詢

方法一

$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

當前命令可能沒有任何返回，別急，試一試下面的命令：

方法二

$ cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

若方法一、方法二都失效了，你還安裝了cudnn？OK，進入/usr/local/cuda/include/下查看上面兩個文件，打開文件如果內容明顯很少，可能是解壓的時候出現問題！

方法三

$ cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2

安裝的路徑有可能比較復雜，可以根據自己的情況調整！