1. 程式人生 > >ubuntu 16.04 +nvidia顯示卡驅動+caffe+cudnn安裝過程及問題總結

ubuntu 16.04 +nvidia顯示卡驅動+caffe+cudnn安裝過程及問題總結

本人之前一直在研究經典的人臉檢測問題,也即是haar人臉檢測,但是現在深度學習是大火的時代,所以現在開始轉向深度學習研究,環境搭建是學習第一步,筆者比較菜,總共用了3天才把caffe環境搭建好,當然搭建環境多謝以下博主,文章摘自以下博主,本人僅為了能夠記錄下來以後可以容易翻到,多謝博主。

安裝硬體: intel i7 + NVIDIA 965 M

作業系統:ubuntu16.04

1、安裝依賴包

2、安裝nvidia顯示卡驅動

3、配置環境變數

4、下載 CUDA 8.0

5、安裝 CUDA 8.0

6、驗證 CUDA 8.0 是否安裝成功

7、安裝 cudnn

8、安裝 opencv3.1

9、安裝 caffe

10、安裝 pycaffe notebook 介面環境

1、安裝依賴包

安裝後續步驟或環境必需的依賴包,依次輸入以下命令:

首先

sudo apt-get update 先對系統軟體包進行一下更新,防止後續無法更新。

然後執行以下幾個步驟

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 
sudo apt-get install git cmake build-essential

2、安裝nvidia顯示卡驅動

上面是nvidia顯示卡驅動下載官方網站,找尋自己的顯示卡型別,如下圖所示,注意下載下來是.run結尾的檔案

由於驅動名字較長,將其更名為a.77.run放在home資料夾下,

2.1 禁用nouveau

sudo gedit /etc/modprobe.d/blacklist.conf

在最後新增:

blacklist nouveau

之後,執行命令:

sudo update-initramfs -u

電腦重啟之後執行 lsmod | grep nouveau

#沒有輸出,即說明安裝成功

2.2 安裝驅動

後續更新中。。。