禁用nouveal自帶驅動(可選,推薦)
控制檯輸入命令,創建一個文件通過命令
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
並添加如下內容:
blacklist nouveau
options nouveau modeset=0
更新一下:
sudo update-initramfs -u
有些博客上說要重啓,此處建議不要重啓,避免出現不能進入系統的問題.
安裝GPU驅動
Ctrl+Alt+F1進入控制檯,輸入以下命令:
sudo service lightdm stop
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get updates
sudo apt-get install nvidia-384
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
sudo reboot
其中 ‘nvidia-384’ 是驅動版本,上官網查詢對應GPU型號的驅動即可。通過 ‘nvidia-smi’ 命令可以查看驅動是否正確安裝。重啓後可以正常使用圖形界面。如果重啓後分辨率有問題,請參考這裏
CUDA安裝
首先安裝缺少的依賴包(強烈推薦,避免之後的安裝出問題)
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libgl1-mesa-dev libglu1-mesa libglu1-mesa-dev libxi-dev
安裝執行文件
從英偉達驅動官網下載對應的cuda驅動程序,如cuda8.0,選擇 ‘run’ 可執行文件下載。下載完成後輸入命令安裝:
sudo sh cuda8.0***.run
首先是大段大段的協議,一直按 NETER 即可,注意 “more(%)”。選擇 “accept“ 接受,接下來根據提示安裝。==注意:當詢問是否安裝 cuda toolkit, cuda-exampls 時選擇yes, 詢問是否安裝其他驅動時一定要選擇no。==其餘如安裝路徑等默認即可
設置環境變量,輸入命令:
sudo gedit ~/.bashrc
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda
sudo source ~/.bashrc
sudo ldconfig
檢測cuda是否配置正確。輸入命令:
‘nvcc - -version’ ,如果出現cuda版本信息等則說明安裝正確。
還可以繼續測試cuda的samples.參考這裏
安裝cudnn
首先到官網下載對應版本的cudnn。安裝cudnn比較簡單,簡單地說,就是複製幾個文件:庫文件和頭文件。將cudnn的頭文件複製到cuda安裝路徑的include路徑下,將cudnn的庫文件複製到cuda安裝路徑的lib64路徑下。具體操作如下:
tar -zxvf cudnn-8.0-linux-x64-v7.tgz
cd cuda
(就是剛剛解壓出來的文件夾)
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7
(#刪除原有動態文件,版本號注意變化,可在cudnn的lib64文件夾中查看)
sudo ln -s libcudnn.so.7.0.2 libcudnn.so.7
(生成軟銜接(注意這裏要和自己下載的cudnn版本對應,可以在/usr/local/cuda/lib64下查看自己libcudnn的版本))
sudo ln -s libcudnn.so.7 libcudnn.so
sudo ldconfig -v
(立刻生效)
查看安裝cudnn後cuda是否依舊可用:
nvcc --version
這樣,cudnn就安裝完成了。如果要進一步驗證,請參考這裏
參考鏈接
和前面的有重複: