ubuntu16.04安裝英偉達驅動,cuda8.0和cudnn

禁用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就安裝完成了。如果要進一步驗證,請參考這裏

參考鏈接

  和前面的有重複:

1

2

3

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章