编译caffe是一个比较复杂的过程,我选择的是Ubuntu16.04和CUDA8.0,都是比较新的版本。
参考链接:
Caffe官方链接,官方针对Ubuntu16.04的安装教程给的非常快,也给出了一些问题的解决方法。
CUDA安装
CUDA 8.0目前需要注册Nvidia的账号才能下载,我选择的是X86_64的.run文件格式。安装CUDA的过程中需要先将图形界面关掉。ctrl+alt+F1切换到命令行模式。输入:
sudo service lightdm stop
切换到刚才下载的.run文件所在的路径中
sudo cuda_8.0.27_linux.run.run
在选择的过程中,首先要输入accept,然后有三个选项,第一个选项我选的是no.即不安装显卡驱动,(因为显卡驱动已经有了,并且这里的驱动比较老-361)第二个选项才是是否安装cuda_toolkit,我选的Yes.第三个选项是是否安装cuda例子,我没有选。
安装成功后,还需要设置环境变量:
设置方法:
sudo vim ~/.bashrc
在文件末尾,添加:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PYTHONPATH=/home/grayson/caffe/python:$PYTHONPATH
保存后在终端输入:
source ~/.bashrc
输入nvcc–version就可以看到CUDA的版本.说明CUDA已经配置好了。