linux安裝GPU顯示卡驅動、CUDA和cuDNN庫
阿新 • • 發佈:2019-02-05
搞機器學習也有一段時間了,每次部署GPU開發環境就是一場戰爭,先記錄一下基本步驟,結合網上資料和個人實踐整理如下:
1、檢查BIOS啟動項,關閉一些選項
在開機啟動項的Security選項中檢查UEFI是否開啟,如果開啟的話請立馬關掉它(重要)
在開機啟動項的Boot選項中檢查Secure Boot是否開啟,如果開啟的話請立馬關掉它(重要)
2、安裝相關依賴
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
如果有圖形化介面則需要禁用x-window服務
sudo service lightdm stop
或
sudo /etc/init.d/lightdm stop
3、禁用 nouveau
安裝好依賴包後需要禁用 nouveau,只有在禁用掉 nouveau 後才能順利安裝 NVIDIA 顯示卡驅動,禁用方法就是在 sudo vim /etc/modprobe.d/blacklist.conf檔案中新增一條禁用命令
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
執行如下命令:
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf build the new kernel by:
最後更新後重啟電腦
sudo update-initramfs -u
reboot
4、查詢NVDIA驅動
首先去官網 http://www.nvidia.com/Download/index.aspx?lang=en-us
下載完後安裝即可,之後就能用nvidia-smi命令了
5、安裝CUDA庫
進入 https://developer.nvidia.com/cuda-downloads,依次選擇 CUDA 型別然後下載對應的CUDA即可,下面以CUDA8.0為例:
如果有圖形化介面則需要禁用x-window服務
安裝cuda驅動
sh cuda_8.0.61_375.26_linux.run
注:
這個時候會有很多提示需要你確認,由於在第一步已經成功安裝了顯示卡驅動,所以這裡就不需要再次安裝了,因此 ,Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 3XX.XX ?這裡選擇 no,其餘都是yes或者accept
在.bashrc新增環境變數後source .bashrc
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
6、驗證CUDA是否安裝成功
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
若看到類似以下資訊則說明 cuda 已安裝成功:
7、安裝cuDNN
下載後進行解壓:
tar -zxvf cudnn-8.0-linux-x64-v5.0-ga.tgz
進入cuDNN5.0解壓之後的include目錄,在命令列進行如下操作:
cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include #複製標頭檔案
再將進入lib64目錄下的動態檔案進行復制和連結:
cd ..
cd lib64
sudo cp lib* /usr/local/cuda/lib64/ #複製動態連結庫
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.5.0.5
sudo ln -sf libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig