關於顯卡驅動安裝在另一篇博客裏,這裏介紹Cuda+cudnn的安裝
本人初學,教程中如有紕漏歡迎指正
如果需要安裝,建議大家多讀幾篇博客作爲參考,可以有效避免他人遇到的問題
如參考本篇文章,一定通讀全文後再做嘗試
一、首先官網查看cuda和顯卡驅動版本對應關係:
(如果做學習研究用,最好提前看好自己所用架構是否能支持所選cuda版本)
二、Cuda下載地址
單擊左側Cuda版本,跳轉後依次選擇操作系統-計算機位數-系統型號-系統版本(由於Ubuntu系統截圖修改不便,語言描述吧)
三、安裝,最終參考官網吧,一步一步來安裝說明
(不同版本官網說明不同,自行查找)
1.查看GPU信息:lspci | grep -i nvidia
2.查看系統信息:uname -m && cat /etc/*release
3.查看GCC信息:gcc --version
4.往下滑,找到.run文件的安裝說明(Runfile Installation):
直接運行下載的.run文件安裝即可:sudo sh cuda_10.1.243_418.87.00_linux.run
如果提示/tmp空間已滿,莫慌,百度尋找解決辦法,這裏不在額外提供
繼續,進入下一步
輸入accept
回車
取消掉Driver的勾選,安裝其他的,Install回車,稍等安裝結果如下:
會提示缺少命令庫,不知道是否有影響,不放心的可以用如下命令安裝庫後,重新安裝一下Cuda:sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
此時結果如下:
配置一下環境變量:sudo gedit ~/.bashrc
加入一下內容,並保存:
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
進入如下目錄:cd /usr/local/cuda/samples/1_Utilities/deviceQuery
執行:sudo make
執行:sudo ./deviceQuery
四、安裝Cudnn
1.下載地址官網
需要註冊賬號後登錄,纔可以下載
下載完成後,逐條執行一下命令(解壓-複製文件到制定目錄):
tar -zxvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
進行驗證,執行cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
結果如下,表示安裝完成
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
如需卸載,刪除複製的文件即可,命令如下:
rm rf命令一定要小心謹慎使用
sudo rm -rf /usr/local/cuda-9.0/lib64/libcudnn*
sudo rm -rf /usr/local/cuda-9.0/include/cudnn.h
到此,Cuda 和Cudnn安裝完成