【Caffe2】Windows7下caffe2编译配置全记录(GPU版)

本文详细记录了在Windows7环境下,配置Caffe2(GPU版)的完整步骤,包括CUDA、cuDNN、Python、Visual Studio、CMake的安装,以及Caffe2源码的获取、编译过程。在编译过程中,可能出现的问题及解决方案也一并给出,例如cub头文件缺失、编译选项设置等。最后,测试了编译后的Caffe2并对比了与Linux环境下的性能差异。
摘要由CSDN通过智能技术生成

caffe2跟老版caffe相比变化较大,更容易上手和使用。但caffe2的windows版配置比linux下要麻烦一些、且更容易出错,把全过程记录下,以供参考。


1. 准备工作

准备工作与Caffe和TF等框架类似,大部分人可以选择性跳过~

(1)安装CUDA、cuDNN

【CUDA下载地址】https://developer.nvidia.com/cuda-downloads
【cuDNN下载地址】https://developer.nvidia.com/cudnn
本文采用的是CUDA8.0+cuDNN7,把cuDNN的lib和include下文件分别复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下的对应文件夹。

(2)安装Python

caffe2支持Python 2.7.6 - Python 2.7.14。
推荐Anaconda安装,一步到位:
【Anaconda下载地址】https://www.anaconda.com/download/#windows

(3)安装Visual Studio

本文采用的是Visual Studio 2015,VS2017也可以。
【Visual Studio 下载地址】https://www.visualstudio.com/zh-hans/downloads/

(4)安装Cmake

【Cmake下载地址】https://cmake.org/download/

(5)安装依赖库

添加python路径到环境变量中,如 C:\Python27 和 C:\Python27\scripts

安装必需的python依赖:

pip install future ^
            hypothesis ^
            numpy ^
            protobuf ^
            six

非必选的python依赖,用于运行caffe2自带的tutorials:

pip install flask ^
            glog ^
            graphviz ^
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值