Code Monkey home page Code Monkey logo

caffe's Introduction

Windows Installation

This is not the original Caffe Readme but an installation guide for windows version.

Want to run first before build by yourself?

You can download the windows x64 standalone package and run directly on MNIST dataset.

Prerequisites

You may need the followings to build the code:

  • Windows 64-bit
  • MS Visual Studio 2012
  • CUDA toolkit 6.5
  • Other dependencies which you can directly download from here.

Build Steps

Currently it can be built by VS2012 for x64 flatform only. This is because the dependencies mentioned above is cross-compiled to support x64 only. If you want to build on 32bit windows, you need to rebuild your own 3rd-party libraries.

  • Check out the code and switch to windows branch
  • Download the dependency file and extract the folders inside to project root directory.
  • Include any .cpp you want to build in the ./tools directory to MainCaller.cpp.
  • Open the solution file in ./build/MSVC
  • Switch build target to x64 platform (Both debug and release are OK).
  • Build the code and you may find the ./bin/MainCaller.exe

Train MNIST dataset

  • Suppose you choose to build train_net.cpp which is the default one in MainCaller.cpp
  • If you do not have GPU, please change it to CPU in lenet_solver.prototxt
  • Goto directory ./examples/mnist
  • Double click get_mnist_leveldb.bat to download the dataset in leveldb format.
  • Double click train_lenet.bat to see the training progress .

Tips

  • It takes obvious longer time when you compile for the first time. Therefore please refrain from using clean & rebuild.
  • To support different GPU compute capabilities, the code is built for several compute capability versions. If you know the exact version of your GPU device, you may remove the support to other versions to speed up the compiling procedure. You may wish to take a look at #25 for more details.

Known Issues

  • I have trained on ImageNet with this windows porting as well. The speed is much slower than the one built on Ubuntu. 20 iterations take 79s on Windows, whereas same number of iterations take about 30s on Ubuntu (on GTX Titan).
  • The above issue has been solved since the upgrade of GPU driver to 340.62 and CUDA to 6.5. The same hardware is now running 26 sencods for 20 iterations on Windows. Hooray!!

Bug Report

  • Please create new issues in github if you find any bug.
  • If you have new pull requests, they are very welcome.
  • Before you do that, you may wish to read this wiki page for more information.

caffe's People

Contributors

yangqing avatar jeffdonahue avatar shelhamer avatar kloudkl avatar sergeyk avatar longjon avatar sguada avatar rbgirshick avatar niuzhiheng avatar mavenlin avatar jamt9000 avatar erictzeng avatar tdomhan avatar chyojn avatar viirya avatar aloa avatar yosinski avatar jetpaccomputer avatar rodrigob avatar beam2d avatar blackball avatar forresti avatar t-abe avatar

Watchers

James Cloos avatar  avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.