淘先锋技术网

首页 1 2 3 4 5 6 7

ubuntu16.04 + GPU 1080ti + CUDA8.0环境下配置 Caffe-SSD

主要记录安装的过程和遇到的问题，大部分都是根据网上的教程整合的。

1.首先，安装依赖，在终端中依次执行即可。

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.安装显卡驱动

首先可以输入命令查看自己是否安装了显卡驱动：

nvidia-smi

如果输出以下界面，说明驱动已经装好，可以跳过下面的步骤。

如果驱动没有安装，可以去NVIDIAlspci | grep NVIDIA

接下来可以直接下载官网上的驱动，手动安装。或者直接在终端输入命令安装适合的版本，我这里安装的是nvidia-384。

sudo apt-get install nvidia-384

装好后，可以重新输入nvidia-smi命令来查看是否装好，如果不能显示那个界面可以重启再试试。

3.安装CUDA8.0

去官网上下载安装包：

点击Download下载，或者使用wget命令：

wget https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda_8.0.44_linux-run

下载完成后，执行如下命令：

sudo chmod 777 cuda_8.0.44_linux-run
sudo cuda_8.0.44_linux-run

接下来就开始了安装过程。注意这里可能会报错说gcc版本过高，需要降低gcc版本。一开始是很长很长的协议，按住空格看完之后输入accept，然后依次选择n（不要安装驱动，因为咱们之前已经装好了，如果这里安装可能导致循环重启的问题）y y y。安装完成之后，需要配置环境变量，在终端输入：

vim ~/.bashrc

按i进入编辑模式，在文本最后输入：

export  PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export  LD_LIBRARY_PATH=/usr/local/cuda8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

然后按esc退出编辑模式，输入:wq保存（别忘了输入冒号），最后激活新的环境：

source ~/.bashrc

然后测试示例，

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery 
make 
sudo ./deviceQuery

如果出现以下输出，说明安装成功。

4.安装cuDNN5.1

去官网下载cuDNN，注意cuDNN的版本和CUDA的版本是有对应关系的，一定要下载正确版本的cuDNN。下载解压之后进入，cuda文件夹，然后执行下面的命令：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
cd cuda; sudo cp lib64/lib* /usr/local/cuda/lib64/; 
sudo cp include/cudnn.h /usr/local/cuda/include/  
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.5.1.10
sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig

请注意解压后的lib64/libcudnn.so.5.0.5 ，文件名可能会不同，要进行相应的修改。

5.安装opencv

由于我这台电脑上有opencv了，如果没安装opencv可以自己安装下。

6.配置Caffe-SSD

下载https://github.com/weiliu89/caffe/tree/ssd源码，或者：

git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd

然后需要修改Makefile.config文件，首先复制一份：

cp Makefile.config.example Makefile.config

然后需要修改以下两行：

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/hdf5/serial/

之后需要把caffe的python路径添加到环境变量里：

vim ~/.bashrc

在文本最后输入：

export PYTHONPATH=$CAFFE_ROOT/python # 注意填写自己的caffe路径

保存后激活环境：

source ~/.bashrc

然后执行以下命令：

make -j8
make pycaffe
make test -j8
make runtest -j8

如果最后输出这样的界面，就代表caffe配置成功了：

 

参考资料：https://blog.csdn.net/a_z666666/article/details/72853346

                  https://www.cnblogs.com/even-ctit/p/8182832.html