caffe學習 在ubuntu下安裝caffe

2021-08-27 03:00:42 字數 2235 閱讀 3403

環境及配置:ubuntu14.04 + cuda8.0 + cudnn5,nvidia gtx1080

開啟系統設定》軟體和更新》附加驅動,選擇nvidia的驅動,版本稍微高一點就好,這裡選擇的384版本的驅動,然後點應用更改,等待自動安裝完成。

1.安裝cuda

sudo sh cuda_8.0.61_375.26_linux.run

一直按空格鍵跳過說明,然後中間選擇不安裝375版本的驅動即可(已經裝過384驅動了)

do you accept the previously read eula?

accept/decline/quit: accept

install nvidia accelerated graphics driver for linux-x86_64 375.26?

(y)es/(n)o/(q)uit: n

剩下的一直選擇yes即可

2.安裝cudnn5

還是上網盤鏈結吧: 提取密碼:sird

解壓後開啟terminal輸入:

sudo rm -rf /usr/local/cuda/include/cudnn.h

sudo rm -rf /usr/local/cuda/lib64/libcudnn*

sudo cp cudnn5/include/cudnn.h /usr/local/cuda/include/

sudo cp cudnn5/lib64/lib* /usr/local/cuda/lib64/

cd /usr/local/cuda/lib64/

sudo rm -rf libcudnn.so libcudnn.so.5

sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5

sudo ln -s libcudnn.so.5 libcudnn.so

sudo ldconfig /usr/local/cuda/lib64

use_cudnn := 1

use_leveldb := 1

use_lmdb := 1

with_python_layer := 1

matlab_dir := /home/user/matlab #新增自己的matlab路徑

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

3.在makefile中更改

nvccflags += -ccbin=$(cxx) -xcompiler -fpic $(common_flags)

替換為:

nvccflags += -d_force_inlines -ccbin=$(cxx) -xcompiler -fpic $(common_flags)

執行:

make all -j8

等待編譯完成。

4.編譯matcaffe和pycaffe

make matcaffe

make pycaffe

sudo gedit ~/.bashrc

將下面**新增到最後一行(路徑修改為自己的caffe路徑),儲存並退出

export pythonpath=/home/user/caffe/python:$pythonpath

執行:

source ~/.bashrc

使環境變數生效。

進入python:

import caffe

未報錯說明caffe能正常使用。

caffe在ubuntu16 04下的安裝

sudo apt get install git dev protobuf compiler sudo apt get install no install recommends libboost all dev sudo apt get install libatlas base dev sudo...

ubuntu下caffe安裝python3介面

正常遇到的問題和python2安裝那些個,直接看下面鏈結。try xrange python 2 except nameerror xrange range python 3 1.修改makefile.config中python的路徑 3.python資料夾下面 sudo pip3 install ...

在windows下配置caffe

安裝環境 win7 64bit vs2012 安裝步驟 安裝 cuda6.5 3.解壓caffe原始碼,解壓dependency,分別將dependency根資料夾內三個資料夾的內容分別解壓到caffe中對應的資料夾內 4.執行 bulid mainbulider.sln,將vs的生成物件設定為64...