1. 程式人生 > >ubuntu14.04中tensorflow和caffe的配置

ubuntu14.04中tensorflow和caffe的配置

1:python2.7+cuda8.0+cudnn6.0+tensorflow1.4.0

python2.7依賴庫安裝:

sudo pip install imutils
sudo apt-get install build-essential python-dev python-numpy python-setuptools python-scipy libatlas-dev libatlas3-base 
sudo apt-get install python-matplotlib  

sudo apt-get install build-essential python-dev python-setuptools python-numpy python-scipy libatlas-dev libatlas3gf-base
pip install --user --install-option="--prefix=" -U scikit-learn

cuda8.0,和cudnn6.0和上文一樣的配置方法,這裡就不在多說了,之所以用tensorflow1.4.0一方面是因為除錯一本書的程式碼需要,另一方面則是cuda8.0僅僅支援1.4版本以下的,如果再高則需要cuda9.0版本了.

tensorflow1.4.0安裝很簡單,直接輸入:

sudo pip install --upgrade https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.4.0rc1-cp27-none-linux_x86_64.whl

安裝過程可能出現安裝SSE4.1, SSE4.2, AVX, AVX2, FMA,等等之類的提示,由於安裝的GPU版本的tensorflow,因此忽略這個問題,因為安裝SSE4.1, SSE4.2, AVX, AVX2, FMA, 僅僅提升CPU的運算速度(大概有3倍).

就可以了.終端輸入:

python
import tensorflow as tf

出現下面介面,表示安裝成功

2:ubuntu14.04+opencv3.1+cuda8.0+python2.7+cudnn5.1+caffee

由於依賴的cudnn版本不一致,因此使用的時候記得切換cudnn版本,當然你也可以安裝成相同版本的,免得麻煩.不過由於我的除錯需要,暫時就這樣了.

(1)依賴項安裝

General dependencies:
sudo apt-get install build-essential 

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 libgflags-dev libgoogle-glog-dev liblmdb-dev
BLAS:
sudo apt-get install libatlas-base-dev

pycaffe:
sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb python-networkx python-nose python-pandas python-gflags Cython ipython
sudo apt-get install protobuf-c-compiler protobuf-compiler

(2)opencv3.1+contrib安裝

sudo apt-get install build-essential
sudo apt-get install cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev
git clone https://github.com/Itseez/opencv.git

git clone https://github.com/Itseez/opencv_contrib.git

mkdir build

cd build
cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D OPENCV_EXTRA_MODULES_PATH=../opencv_contrib/modules/ -D CUDA_ARCH_BIN=6.1 -D CUDA_GENERATION=Auto ..

opencv和contrb的路徑自己設定,這個應該很簡單吧,CUDA運算能力必須和自己的GPU匹配,否則會報錯.cudnn就不多說了

python是系統自帶的2.7.6版本.

3:caffe安裝

(2):cp Makefile.config.example Makefile.config 

(3):修改配置檔案Makefile.config,我取消了這兩處的註釋

USE_CUDNN := 1
OPENCV_VERSION := 3

(4)如果編譯出現找不到cuda庫的一些問題,解決如下:

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 -P /usr/local/cuda-8.0/lib64/libcudnn* /usr/local/lib && sudo ldconfig

(5)編譯

make all -j16
make test -j16
make runtest -j16
make pycaffe

(6)測試

1.將終端定位到Caffe根目錄
cd ~/caffe
		 
2.下載MNIST資料庫並解壓縮
./data/mnist/get_mnist.sh

3.將其轉換成Lmdb資料庫格式
./examples/mnist/create_mnist.sh
		 
4.訓練網路
./examples/mnist/train_lenet.sh

正常執行則安裝成功.

(7)使用python匯入caffe

新增環境變數 
export PYTHONPATH=~/lib/caffe:$PYTHONPATH 

還是找不到caffe模組的話,寫個路徑指令碼吧mycaffe.py
#!/usr/bin/python
"""
caffe path
"""
caffe_root = '~/lib/caffe/'
import sys
sys.path.insert(0, caffe_root + 'python')
import caffe  

(8)測試結果如下:

祝大家都能成功安裝,踏上深度學習之路,哈哈!