caffe安装流程和遇到问题

之前caffe的安装简直让我怀疑人生,后来由于忙一直没有写下流程和重要问题的解决办法,这次由于在自家的电脑上配置caffe,顺便下写流程。

不多说,先上自家电脑配置,i5-4590和GTX1070

1.caffe安装依赖库

[plain]  view plain   copy
  print ? 在CODE上查看代码片 派生到我的代码片
  1. sudo apt-get install build-essential  # basic requirement  
  2. sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler #required by caffe  
首先最好一个一个安装,其次这里有可能出现需要依赖库,以下依赖库将不被安装,解决方法是换软件源,我换了清华的源后立马就好了


2.安装CUDA

CUDA是英伟达的显卡并行计算语言,caffe需要来使用显卡

在离线.deb安装:deb安装分离线和在线,官网下载地址,推荐离线安装

切换到下载的deb所在目录,执行下边的命令
[plain]  view plain   copy
  print ? 在CODE上查看代码片 派生到我的代码片
  1. sudo dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb  
  2. sudo apt-get update  
  3. sudo apt-get install cuda  
然后重启电脑:sudo reboot

3.cuDNN
这个是显卡计算加速的,可以不装,加速效果远没有GPU对CPU的加速来的多,而且安装还麻烦
下载cudnn-7.5-linux-x64-v5.0-ga.tgz,官网申请不到,网上自己找的,就不给地址了。
[plain]  view plain   copy
  print ? 在CODE上查看代码片 派生到我的代码片
  1. tar -zxvf cudnn-7.5-linux-x64-v5.0-ga.tgz  
  2. cd cuda  
  3. sudo cp lib/lib* /usr/local/cuda/lib64/  
  4. sudo cp include/cudnn.h /usr/local/cuda/include/  
更新软连接
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
 
 
  4,设置环境变量
在/etc/profile中添加CUDA环境变量
sudo gedit /etc/profile
[plain]  view plain   copy
  print ? 在CODE上查看代码片 派生到我的代码片
  1. PATH=/usr/local/cuda/bin:$PATH  
  2. export PATH  
保存后, 执行下列命令, 使环境变量立即生效
[plain]  view plain   copy
  print ? 在CODE上查看代码片 派生到我的代码片
  1. source /etc/profile  
同时需要添加lib库路径: 在 /etc/ld.so.conf.d/加入文件 cuda.conf, 内容如下
[plain]  view plain   copy
  print ? 在CODE上查看代码片 派生到我的代码片
  1. /usr/local/cuda/lib64  
保存后,执行下列命令使之立刻生效
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值