Caffe与Tensorflow的安装与配置

Caffe配置

1.安装一些基本依赖项：

1
2
3
4
$ 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 libopenblas-dev liblapack-dev libatlas-base-dev
$ sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev  

2.cuda7.5安装

 

1
2
3
$ sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda  

配置环境变量

1
2
$ export PATH=/usr/local/cuda-7.5/bin:$PATH 
$ export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH  

• sudo gedit ~/.bashrc
• #在文档最后添加以下命令
• export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"
• export CUDA_HOME=/usr/local/cuda
• #保存退出后在terminal中执行
• source ~/.bashrc

3. Caffe

git：

一.

1
$ sudo add-apt-repository ppa:git-core/ppa  

1
2
sudo apt-get update
$ sudo apt-get install git  

二.下载caffe

1
$ git clone https://github.com/BVLC/caffe.git  

 

编译caffe：

1
2
3
$ cd ~/caffe
$ sudo cp Makefile.config.example Makefile.config
$ make all  

4.cuDnn

 

1
2
3
4
5
6
7
8
9
10
$ sudo tar xvf cudnn-7.0-linux-x64-v4.0-prod.tgz
$ cd cuda/include
$ sudo cp *.h /usr/local/include/
$ cd ../lib64
$ sudo cp lib* /usr/local/lib/
$ cd /usr/local/lib
$ sudo chmod +r libcudnn.so.4.0.7
$ sudo ln -sf libcudnn.so.4.0.7 libcudnn.so.4
$ sudo ln -sf libcudnn.so.4 libcudnn.so
$ sudo ldconfig  

然后切换到caffe根目录下，将Makefile.config中的USE_CUDNN行前的#去掉：

1
2
$ cd ~/caffe
$ sudo vi Makefile.config  

保存后重新编译：

1
2
3
4
$ sudo make clean
$ sudo make all
$ sudo make test  
$ sudo make runtest   

 

 

本人编译的时候最后一步的时候，出现这样的错误：

1
libcudart.so.7.5 cannot open shared object file: No such file or directory  

网上有一堆介绍，怎么设置环境变量的（我们分明已经设置过了），可以这样检查是否已经添加环境变量：

1
2
3
4
$ echo $PATH
$ echo $LD_LIBRARY_PATH  

解决方法是这样，将一些文件复制到/usr/local/lib文件夹下：

1
2
3
$ sudo cp /usr/local/cuda-7.5/lib64/libcudart.so.7.5 /usr/local/lib/libcudart.so.7.5 && sudo ldconfig
$ sudo cp /usr/local/cuda-7.5/lib64/libcublas.so.7.5 /usr/local/lib/libcublas.so.7.5 && sudo ldconfig
$ sudo cp /usr/local/cuda-7.5/lib64/libcurand.so.7.5 /usr/local/lib/libcurand.so.7.5 && sudo ldconfig  

 

 

 

Tensorflow 配置

安装pip

1
2
3
$ sudo apt-get install python-pip python-dev python-setuptools build-essential
$ sudo pip install --upgrade pip
$ sudo pip install --upgrade virtualenv  

安装Tensorflow  0.8.0

1
2
3
# Ubuntu/Linux 64-bit, GPU enabled. Requires CUDA toolkit 7.5 and CuDNN v4.  For
# other versions, see "Install from sources" below.
$ sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.8.0-cp27-none-linux_x86_64.whl

 

CUDA7.5：https://developer.nvidia.com/cuda-downloads

cuDNN V4：https://developer.nvidia.com/rdp/cudnn-download