caffe+ubuntu18.04+cuda10.2编译配置

本文详细介绍了在Ubuntu18.04系统上,针对CUDA10.2环境如何手动安装cmake 3.14.3并解决编译Caffe时遇到的CUDA库问题。步骤包括安装依赖、设置环境变量、修改Makefile.config以及处理CUDA_ARCH的配置,确保成功编译和运行Caffe。
摘要由CSDN通过智能技术生成

安装相应依赖

apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
apt-get install --no-install-recommends libboost-all-dev
apt-get install python-dev
apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
apt-get install libatlas-base-dev
apt-get install the python-matplotlib python-scipy python-numpy
pip3 install pytest numpy boost

安装CMAKE

注意:使用apt-get安装的cmake版本太低,在使用CUDA10.2的环境下进行caffe编译时会出现CUDA_cublas_device_LIBRARY/CUDA_cublas_LIBRA报错。这是由于低版本cmake找不到cuda的库所导致。为解决这个问题,我们需要手动安装cmake,版本要求至少为3.12.0。我们选择3.14.3。

如果已经存在cmake请确认版本 cmake --version

  1. 从cmake官网下载cmake-3.14.3-Linux-x86_64.tar.gz
  2. 解压 tar -zxvf cmake-3.14.3-Linux-x86_64.tar.gz
  3. 在~/.bashrc文件中添加环境变量,在文件末尾添加:
    export CMAKE_ROOT=/home/shiyh/download/cmake-3.14.3-Linux-x86_64
    export PATH=$PATH:$CMAKE_ROOT/bin:
    注意/home/shiyh/download/请替换为具体路径
  4. 保存,并执行source ~/.bashrc

cmake编译

  1. 修改Makefile.config
    源码文件夹中已提供Makefile.config.example
    执行cp Makefile.config.example Makefile.config
  2. 修改CMakeList.txt中第35行
    第35行的set(python_version “2” CACHE STRING “Specify which Python version to use”)中的2改为3.6(根据自己的python版本确定),保存退出
  3. vim Makefile.config  对配置文件中相关参数修改
    ## Refer to http://caffe.berkeleyvision.org/installation.html
    # Contributions simplifying and improving our build system are welcome!
    
    # cuDNN acceleration switch (uncom
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值