Caffe安装教程(Ubuntu15.04+CUDA7.5)

Caffe安装教程(Ubuntu15.04+CUDA7.5)

Caffe作为开源的深度学习框架,有着代码结构清晰,速度快等优点。但是配置与安装一直是个比较令人头疼的问题。尤其是随着每个Caffe包版本的升级以及CUDA等必要的库的更新换代,之前能用的教程说不定版本一更新就用不了了。在配置Caffe的时候我就走了不少弯路。最后折腾了好几天才搞定。这里我把我配置的流程简单介绍一下。

Ubuntu的安装

我安装的是Ubuntu15.04版本,安装Ubuntu是相对来说比较简单的。你只要需要一个作为启动盘的U盘即可。Ubuntu安装完以后我们优先安装vim和git两个重要的软件。

sudo apt-get update 
sudo apt-get install git vim cmake automake

必要软件安装和环境的简单配置

CUDA7.5

打开CUDA官方下载的链接,进行平台的选择。我选择Ubuntu15.04,然后在Installer Type要选择deb(local),这样我们下载一个大小约为2g左右的文件。

然后在文件下好的目录下打开终端看,运行以下三句话就可以了。网上有很多非常复杂的配置CUDA的教程。那是因为他们用的是runfile安装,这样的方法又繁琐又容易出错。这里我们只要简单用local的deb文件,根据官网的教程走就可以了,没必要走复杂的流程。只需运行以下的三句。

sudo dpkg 
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
首先,在 Ubuntu 20 上安装 Caffe-CUDA 需要满足一些先决条件。请确保你已经安装了 CUDA 和 cuDNN,并且已经正确配置了它们。然后,按照以下步骤进行安装: 1. 安装依赖项: ```shell sudo apt-get update sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev liblmdb-dev ``` 2. 克隆 Caffe 仓库: ```shell git clone https://github.com/BVLC/caffe.git ``` 3. 进入 Caffe 目录: ```shell cd caffe ``` 4. 编辑 Makefile.config 文件: ```shell cp Makefile.config.example Makefile.config nano Makefile.config ``` 确保按照你的配置修改以下行: - `USE_CUDNN := 1` (如果已安装 cuDNN) - `CUDA_DIR := /usr/local/cuda` (CUDA 的安装路径) - `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` (根据你的系统配置修改) 5. 编译 Caffe: ```shell make all -j$(nproc) # 使用所有可用的 CPU 内核 make test -j$(nproc) make runtest -j$(nproc) ``` 6. 安装 Caffe: ```shell sudo make install ``` 7. 配置环境变量: ```shell echo 'export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH' >> ~/.bashrc source ~/.bashrc ``` 现在,你应该已经成功在 Ubuntu 20 上安装了 Caffe-CUDA。你可以尝试运行一些示例代码来验证安装是否成功。记得替换 `/path/to/caffe` 为你的 Caffe 目录路径。希望对你有所帮助!

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值