之前caffe的安装简直让我怀疑人生,后来由于忙一直没有写下流程和重要问题的解决办法,这次由于在自家的电脑上配置caffe,顺便下写流程。
不多说,先上自家电脑配置,i5-4590和GTX1070
1.caffe安装依赖库
首先最好一个一个安装,其次这里有可能出现需要依赖库,以下依赖库将不被安装,解决方法是换软件源,我换了清华的源后立马就好了
2.安装CUDA
CUDA是英伟达的显卡并行计算语言,caffe需要来使用显卡
在离线.deb安装:deb安装分离线和在线,官网下载地址,推荐离线安装
切换到下载的deb所在目录,执行下边的命令
然后重启电脑:sudo reboot
3.cuDNN
这个是显卡计算加速的,可以不装,加速效果远没有GPU对CPU的加速来的多,而且安装还麻烦
下载cudnn-7.5-linux-x64-v5.0-ga.tgz,官网申请不到,网上自己找的,就不给地址了。
更新软连接
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.5.0.5
sudo ln -sf libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig
sudo chmod +r libcudnn.so.5.0.5
sudo ln -sf libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig
4,设置环境变量
在/etc/profile中添加CUDA环境变量
sudo gedit /etc/profile
保存后, 执行下列命令, 使环境变量立即生效
同时需要添加lib库路径: 在 /etc/ld.so.conf.d/加入文件 cuda.conf, 内容如下
保存后,执行下列命令使之立刻生效