主要记录安装的过程和遇到的问题,大部分都是根据网上的教程整合的。
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
如果输出以下界面,说明驱动已经装好,可以跳过下面的步骤。
如果驱动没有安装,可以去NVIDIA官网根据自己的显卡型号,查看适合自己显卡的驱动,查看自己显卡版本的命令:
lspci | 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