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

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 ^
  • 4
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值