1. 程式人生 > >ubuntu安裝caffe+cuda+cudnn+nvidia驅動

ubuntu安裝caffe+cuda+cudnn+nvidia驅動

1、nvidia驅動安裝

先解除安裝原有N卡驅動 sudo apt-get remove --purge nvidia* sudo apt-get --purge remove xserver-xorg-video-nouveau 若已經安裝nvidia驅動,需則要解除安裝 sudo ./NVIDIA-Linux-x86_64-384.59.run --uninstall

下載官方驅動程式   http://www.geforce.cn/drivers(找到自己顯示卡所需要的版本)

禁止整合的nouveau驅動   修改屬性  sudo chmod 666  /etc/modprobe.d/blacklist.conf sudo vim /etc/modprobe.d/blacklist.conf 在該檔案後新增一下幾行: blacklist vga16fb blacklist nouveau blacklist rivafb blacklist rivatv blacklist nvidiafb

開始安裝 關閉當前圖形環境 Ctrl + Alt + F1 sudo service lightdm stop 給驅動run檔案賦予執行許可權: sudo chmod +x NVIDIA-Linux-x86_64-384.59.run(對應自己所下載的驅動版本) 後面的引數非常重要,不可省略: sudo ./NVIDIA-Linux-x86_64-384.59.run –no-opengl-files 最後重新啟動圖形環境 Ctrl + Alt + F7

檢視顯示卡驅動版本  cat /proc/driver/nvidia/version 若列出GPU的資訊列表,表示驅動安裝成功  nvidia-smi 若彈出設定對話方塊,亦表示驅動安裝成功  nvidia-settings

2、CUDA安裝

下載官方cuda https://developer.nvidia.com/cuda-downloads (如cuda_8.0.61_375.26_linux.run)

sudo sh cuda_8.0.44_linux.run --override 啟動安裝程式,鍵入q可跳過條款目錄,輸入accept接受條款 輸入n不再安裝nvidia驅動,之前已經安裝過了   (除了這裡輸入n,其他都是y確認通過) 輸入y安裝cuda 8.0工具 回車確認cuda預設安裝路徑:/usr/local/cuda-8.0 輸入y用sudo許可權執行安裝,輸入密碼 輸入y或者n安裝或者不安裝指向/usr/local/cuda的符號連結 輸入y安裝CUDA 8.0 Samples,以便後面測試 回車確認CUDA 8.0 Samples預設安裝路徑,該安裝路徑測試完可以刪除

新增環境變數     sudo vim ~/.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}}

然後設定環境變數和動態連結庫    sudo vim /etc/profile export PATH=/usr/local/cuda/bin:$PATH 使生效    sudo ldconfig

編譯並測試裝置 deviceQuery: cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery sudo make ./deviceQuery

3、cudnn安裝

下載官方相應的cudnn包,並解壓 https://developer.nvidia.com/cudnn

cd cuda/include/ sudo cp cudnn.h /usr/local/cuda/include/    #複製標頭檔案 sudo cp ../lib64/lib* /usr/local/cuda/lib64/    #複製動態連結庫 cd /usr/local/cuda/lib64/ sudo rm -rf libcudnn.so libcudnn.so.7    #刪除原有動態檔案 sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7  #生成軟銜接 sudo ln -s libcudnn.so.7 libcudnn.so      #生成軟連結

4、opencv安裝

見opencv3.2的安裝教程

5、caffe安裝

安裝依賴庫 sudo apt install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler

sudo apt install -y --no-install-recommends libboost-all-dev libgflags-dev libgoogle-glog-dev liblmdb-dev python-numpy

sudo apt install -y python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb

sudo apt install -y python-networkx python-nose python-pandas python-gflags cython ipython libatlas-base-dev

更新  sudo apt-get update

下載官方caffe https://github.com/BVLC/caffe.git 進入 caffe 將 Makefile.config.example 檔案複製一份並更名為 Makefile.config sudo cp Makefile.config.example Makefile.config 開啟檔案  sudo vim Makefile.config 取消以下行註釋 USE_CUDNN := 1 OPENCV_VERSION := 3 (如果使用opencv3) WITH_PYTHON_LAYER := 1

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib  /usr/lib 修改為: INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include  /usr/include/hdf5/serial LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib  /usr/lib /usr/lib/x86_64-linux-gnu  /usr/lib/x86_64-linux-gnu/hdf5/serial

編譯 make all -j8 測試 sudo make runtest -j8