淘先锋技术网

首页 1 2 3 4 5 6 7

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

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