windows安装caffe

caffe windows上有教程,但是有几点没有详细说明。

1. 安装cuda, cudnn和vc编译器

下载cudacudnn
安装vc编译器后,在cmakelist.txt中设置使用vc编译器而不是gcc:

set(CMAKE_C_COMPILER "D:/Microsoft Visual Studio 12.0/VC/bin/cl.exe")
set(CMAKE_CXX_COMPILER "D:/Microsoft Visual Studio 12.0/VC/bin/cl.exe")

2. 安装boost

boost官网上下载exe安装文件。解压安装完成后需要添加3个环境变量:

BOOST_ROOT=path\to\boost
BOOST_INCLUDEDIR=path\to\boost
BOOST_LIBRARYDIR=path\to\boost\lib64-msvc-14.2

3. 安装cmake

cmake官网上下载cmake并安装。注意将cmake添加到环境变量。

注意:boost和cmake版本必须要匹配,否则在用cmake编译时会提示找不到boost: Could not find the following Boost libraries: boost_system。此处我使用的版本是: cmake-3.17.0, boost-boost_1_72_0

4. 关闭ninja

官方文档中建议用Ninja来加快速度,但是我系统安装ninja后报错,因此在scripts\build_win.cmd设置WITH_NINJA=0。如果提示找不到python或者python版本不对(只支持python2.7python3.5),可以在scripts\build_win.cmd中加上一行:

-DPYTHON_EXECUTABLE:FILEPATH=D:\Python27\python.exe ^

设置python位置:

cmake -G"!CMAKE_GENERATOR!" ^
      -DBLAS=Open ^
      -DCMAKE_BUILD_TYPE:STRING=%CMAKE_CONFIG% ^
      -DBUILD_SHARED_LIBS:BOOL=%CMAKE_BUILD_SHARED_LIBS% ^
      -DBUILD_python:BOOL=%BUILD_PYTHON% ^
      -DBUILD_python_layer:BOOL=%BUILD_PYTHON_LAYER% ^
      -DBUILD_matlab:BOOL=%BUILD_MATLAB% ^
      -DCPU_ONLY:BOOL=%CPU_ONLY% ^
      -DCOPY_PREREQUISITES:BOOL=1 ^
      -DINSTALL_PREREQUISITES:BOOL=1 ^
      -DUSE_NCCL:BOOL=!USE_NCCL! ^
      -DCUDA_ARCH_NAME:STRING=%CUDA_ARCH_NAME% ^
	  -DPYTHON_EXECUTABLE:FILEPATH=D:\Python27\python.exe ^
      "%~dp0\.."

python版本不对会报错:Could not find url for MSVC version = 1800 and Python version = 3.7.

5. 添加环境变量

python\to\caffe\build\tools\Release添加到path环境变量。将python\to\caffe\python添加到PYTHONPATH环境变量。


编译中的其他环境可以在build_win.cmd中修改。

6. 安装graphviz

下载graphviz并安装,然后将path\to\graphviz/bin添加到环境变量。最后运行如下命令来注册插件:

dot -c

这样就可以生成prototxt的网络图了。或者在github中输入网络图查看。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值