caffe + mkl2018 + cuda9.1 + cudnn7.0.5 @ windows vs2015

新年编译。

新年伊始发现mkl、cuda、cudnn都有了新版本,于是在笔记本上重新编译了一下caffe。

首先还是官网下载最新版本的mkl、cuda、cudnn对应的windows平台版本并安装。

cudnn要解压到cuda工具的安装目录方便使用(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1)。

caffe checkout到windows branch。

编译环境我一直用的是vs2015,另外装一个miniconda(python环境)。这次miniconda安装也碰到个坑,安装时一定要选“只有我”,不然conda install需要管理员权限。另外最好选择添加系统路径,不然总是找不到conda.exe。

安装好miniconda后,就可以用anaconda的conda install安装ninja和cmake,用来编译caffe。

编译要在caffe文件夹的cmd下运行scripts\build_win.cmd,注意要修改这个文件把需要的mkl以及cuda配好。

    :: Change the settings here to match your setup
    :: Change MSVC_VERSION to 12 to use VS 2013, 14 to use VS 2015
    if NOT DEFINED MSVC_VERSION set MSVC_VERSION=14
    :: Change to 1 to use Ninja generator (builds much faster)
    if NOT DEFINED WITH_NINJA set WITH_NINJA=1
    :: Change to 1 to build caffe without CUDA support
    if NOT DEFINED CPU_ONLY set CPU_ONLY=0
    :: Change to generate CUDA code for one of the follow
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值