溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶(hù)服務(wù)條款》

Linux下怎么安裝卷積神經(jīng)網(wǎng)絡(luò)框架caffe

發(fā)布時(shí)間:2022-02-12 16:37:54 來(lái)源:億速云 閱讀:227 作者:iii 欄目:開(kāi)發(fā)技術(shù)

這篇文章主要介紹了Linux下怎么安裝卷積神經(jīng)網(wǎng)絡(luò)框架caffe的相關(guān)知識(shí),內(nèi)容詳細(xì)易懂,操作簡(jiǎn)單快捷,具有一定借鑒價(jià)值,相信大家閱讀完這篇Linux下怎么安裝卷積神經(jīng)網(wǎng)絡(luò)框架caffe文章都會(huì)有所收獲,下面我們一起來(lái)看看吧。

caffe是一個(gè)清晰而高效的深度學(xué)習(xí)框架,是純粹的C++/CUDA架構(gòu),支持命令行、python和matlab接口;可以在CPU和GPU之間無(wú)縫切換,其作者是賈揚(yáng)清,目前caffe2已經(jīng)發(fā)布。

Linux下怎么安裝卷積神經(jīng)網(wǎng)絡(luò)框架caffe

下載caffe并保存到一個(gè)目錄下(推薦放到 /home 目錄)

  1. 安裝依賴(lài)項(xiàng):
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler(這一大串是一個(gè)命令)
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install python-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install python-lmdb
sudo apt-get install python-pip
1234567
  1. cd caffe,并修改配置文件
cp Makefile.config.example Makefile.config(如果用CPU跑,把這兩個(gè)文件的cuda相關(guān)都注釋掉)
1
  1. 修改需編譯的選項(xiàng)(針對(duì)ubuntu15.04以及之后的版本,之前的版本不用修改,用于解決無(wú)法找到hdf5.h等文件)
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include \
  /usr/lib/x86_64-linux-gnu/hdf5/serial/include

LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib \
 /usr/lib/x86_64-linux-gnu/hdf5/serial

123456

如果第4步make runtest有錯(cuò)誤,error while loading shared libraries: libcudart.so.8.0: cannot open shared object file: can not open(/usr/local/lib/目錄下找不到libcudart.so.8.0這個(gè)文件, 該文件可以在usr/local/cuda-8.0/lib64/目錄下找到),解決辦法:

sudo cp /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/local/lib/libcudart.so.8.0 && sudo ldconfig
sudo cp /usr/local/cuda-8.0/lib64/libcublas.so.8.0 /usr/local/lib/libcublas.so.8.0 && sudo ldconfig  
sudo cp /usr/local/cuda-8.0/lib64/libcurand.so.8.0 /usr/local/lib/libcurand.so.8.0 && sudo ldconfig
sudo cp /usr/local/cuda-8.0/lib64/libcudnn.so.6 /usr/local/lib/libcudnn.so.6 && sudo ldconfig
1234

或者可以在LIBRARY_DIRS后添加這個(gè)路徑?usr/local/cuda-8.0/lib64/(沒(méi)試過(guò),應(yīng)該可以)

參考鏈接:https://blog.csdn.net/qq_38451119/article/details/81126692

去掉CPU—only選項(xiàng)前的注釋,使之生效,若需Python layer,則還需去掉WITH_PYTHON_LAYER :=1前的注釋 如需編譯MATLAB接口,則需修改MATLAB_DIR,改成當(dāng)前系統(tǒng)下的MATLAB路徑

  1. 編譯caffe:
make all -j4
make test -j4
make runtest
123

編譯python接口

sudo pip install -r python/requirements.txt
make pycaffe
12

編譯matlab接口

make matcaffe
gedit ~/.bashrc
12

最后面添加:export PYTHONPATH=/home/francis/caffe/python

rm *~ 可考慮刪除因修改而自動(dòng)生成的備份文件 以上可成功配置好cpu_only

安裝GPU版本 應(yīng)先裝好CUDA,條件允許可以可加入cudnn,用于加速卷積運(yùn)算 CUDA安裝步驟待后續(xù)完善 注釋掉CPU-only,其他與前面相同。

關(guān)于“Linux下怎么安裝卷積神經(jīng)網(wǎng)絡(luò)框架caffe”這篇文章的內(nèi)容就介紹到這里,感謝各位的閱讀!相信大家對(duì)“Linux下怎么安裝卷積神經(jīng)網(wǎng)絡(luò)框架caffe”知識(shí)都有一定的了解,大家如果還想學(xué)習(xí)更多知識(shí),歡迎關(guān)注億速云行業(yè)資訊頻道。

向AI問(wèn)一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI