caffe 環境搭建和minist訓練

2021-07-14 13:46:18 字數 2483 閱讀 5812

2.編譯convert_mnist_data工程 (用於之後的訓練資料轉換)

獲取到t10k-images-idx3-ubyte、t10k-labels-idx1-ubyte、train-images-idx3-ubyte、train-labels-idx1-ubyte(注意檔案大小)

4.構造mnist資料 參考example/mnist/creae_mnist.sh (window下改為creae_mnist.bat)

此步驟要保證convert_mnist_data工程以及成功編譯,且生成convert_mnist_data.exe(linux下為convert_mnist_data.bin)

執行上訴指令碼後,生成mnist_test_lmdb和mnist_train_lmdb資料夾

5.開啟caffe工程,配置屬性->除錯->命令引數    --solver=../../examples/mnist/lenet_solver.prototxt (訓練引數,而不是網路結構)

6.這裡需要注意的是,新mnist為lmdb格式,而老的是levelmdb格式,需要在lenet_train_test.prototxt中進行修改:

backend: levelmdb  ->  backend: lmdb

1. git clone  caffe

2. mv makefile.config.example makefile.config(cpu)

makefile.config blas := atlas -> blas := open 

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. apt-get install libprotobuf-dev 

apt-get install libboost-dev

apt-get install libgflags-dev

apt-get install libgoogle-glog-dev

apt-get install libleveldb-dev

apt-get install liblmdb-dev

apt-get install libhdf5-dev

apt-get install libboost-system1.55-dev

apt-get install libboost-filesystem1.55-dev

apt-get install libboost-thread1.55-dev

apt-get install libatlas-base-dev   

4.make all -j4

make test -j4

make runtest -j4

make pycaffe -j4

5.data/mnist/get_mnist.sh

gunzip gz檔案

examples/mnist/create_mnist.sh(生成mnist-train-lmdb 和 mnist-train-lmdb資料夾)

修改protxt 中gpu

examples/mnist/train_lenet.sh(訓練mnist)

訓練完後生成

./examples/mnist/lenet_iter_10000.solverstate

./examples/mnist/lenet_iter_10000.caffemodel

6.測試

./build/tools/caffe test

-model=examples/mnist/lenet_train_test.prototxt

-weights=examples/mnist/lenet_iter_10000.caffemodel

-gpu=0

1.hdf5 (error)

tar jxvf hdf5-1.10.1.tar.bz2

./configure --prefix=/home/***_name/hdf5/

make

make install

2.cuda(非root)

./cuda_7.0.28_linux.run

在安裝過程中 設定路徑為

/home/***_name/cuda-7.0

Caffe 環境搭建

一 安裝brew 登入直接複製官網提供命令即可安裝 usr local cudacudnn 7.0 osx x64 v4.0 rc.tar安裝 downloads tar xvf cudnn 7.0 osx x64 v4.0 rc.tar x cuda include cudnn.h x cuda ...

mac下搭建caffe環境

第一步 安裝homebrew。它是mac下的一種包管理工具,用於快速安裝各種包,類似於linux下的yum和python下的pip。ruby e curl fssl 第二步 利用homebrew安裝caffe的各項依賴包 brew install vd leveldb brew install vd...

DPDK環境搭建和測試

gnumake coreutils cmp,sed,grep,arch gcc versions 4.5.x libcheaders glibc devel.i686 libc6 dev i386 glibc devel.x86 64 for 64 bitcompilation linuxkerne...