anaconda 安装caffe 的编译过程

本文档详细介绍了在Anaconda Python 3.5.2、CUDA 8.0、cuDNN v5环境下编译Caffe的全过程,包括安装Visual Studio 2015、CMake 3.4.3、设置环境变量、解决编译问题如找不到pthread.h,并最终测试import caffe是否成功。编译过程中提到了解决"The C/CXX compiler identification is unknown"问题,以及如何配置CUDA和cuDNN路径。最后,确保在Anaconda环境中可以正常使用caffe。
部署运行你感兴趣的模型镜像

Requirements

  • Visual Studio 2013 or 2015
    • Technically only the VS C/C++ compiler is required (cl.exe)
  • CMake 3.4 or higher (Visual Studio and Ninja generators are supported)

Optional Dependencies

  • Python for the pycaffe interface. Anaconda Python 2.7 or 3.5 x64 (or Miniconda)
    • Matlab for the matcaffe interface.
  • CUDA 7.5 or 8.0 (use CUDA 8 if using Visual Studio 2015)
  • cuDNN v5

We assume that cmake.exe and python.exe are on your PATH.

我的安装环境为VS 2015、CMake 3.4.3、Anaconda Python 3.5.2、CUDA 8.0、cuDNN v5
Anaconda和Python的版本对应关系及下载地址可以从这里找到,Anaconda Python

C:\Projects> git clone https://github.com/BVLC/caffe.git
C:\Projects> cd caffe
C:\Projects\caffe> git checkout windows
:: Edit any of the options inside build_win.cmd to suit your needs
C:\Projects\caffe> scripts\build_win.cmd
  • 1
  • 2
  • 3
  • 4
  • 5

我这里使用的是git bash,相应的安装、使用教程可以从这里找到,GitBash
9fa981863dd422b6483dc94851e38978.pnguploading.4e448015.gif转存失败重新上传取消在这里插入图片描述

编译

而后编辑在caffe\scripts路径下的build_win.cmd文件。

  • 如果用CUDA,就SET CPU_ONLY=0
  • PYTHON_VERSION设置自己需要的2或3
  • SET WITH_NINJA=0
  • 在倒数第二行加入CUDA路径
-DCUDNN_ROOT=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 ^
  • 1

编译前请删除,如下文件

caffe\scripts\build路径下的CMakeCache.txt文件
  • 1

若出现如下问题,请首先给VS安装C++组件
–The C compiler identification is unknown
–The CXX compiler identification is unknown

安装完成C++组件后,再次运行build_win.cmd,开始下载
55ee99fc3ce0d67339d069a8b8114a6f.pnguploading.4e448015.gif转存失败重新上传取消在这里插入图片描述
如有提示,找不到pthread.h,这是由于这个库是LINUX、UNIX的VS有的库,而Windows没有,可以参照这个博客进行安装,VS2015配置pthread.h
重新build,经过半个多小时,超多的warning,还是build成功了
ddcc040979f71719883ce337996213d2.pnguploading.4e448015.gif转存失败重新上传取消在这里插入图片描述
那个DCUDNN_ROOT的问题,其实是这样的。
在build_win.cmd里,如果CUDA设置了环境变量,且cuDNN拷进了CUDA,就不需要再指定路径了。
如果CUDA没设置环境变量,就需要指定路径。

我这里是写多余了。

测试

进入/caffe/python路径,打开python,测试import caffe成功
2e6af698b528f1abc100bb08e45ab2b5.pnguploading.4e448015.gif转存失败重新上传取消在这里插入图片描述
将/caffe/python/caffe文件夹拷贝到\Anaconda3\Lib\site-packages目录下,这样直接打开系统默认的Anaconda的Python也可以import caffe了。

用VS2015打开/caffe/scripts/caffe目录下的caffe.sln解决方案
472afc13ee4db34b267be8b7a77dbc5f.pnguploading.4e448015.gif转存失败重新上传取消在这里插入图片描述
对整个解决方案进行Build操作。
等编译完成后,在如下路径会产生caffe.exe文件

C:\Documents\Caffe\caffe\scripts\build\tools\Release
  • 1

事实上不编译这个项目,也会有这个文件,之前的build_win.cmd已经完成了这一步编译…
将路径加入系统变量path,然后在cmd测试。

path='C:\Documents\Caffe\caffe\scripts\build\tools\Release'
cmd
输入 
caffe.exe -version
输出
caffe.exe version 1.0.0

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值