Using Caffe On The Lab Computers

Revision as of 15:17, 10 May 2016 by Steve410.cs.jhu.edu (talk | contribs) (Created page with "<pre> reference CaffeNet model using your Jupyter notebook code, so you should be good to go. There are a couple of things you need to be aware of: * You don't need to `im...")
(diff) ← Older revision | Latest revision (diff) | Newer revision → (diff)

 reference CaffeNet model using your Jupyter
notebook code, so you should be good to go.  There are a couple of
things you need to be aware of:

 * You don't need to `import sys` and add things to `sys.path`.  You can
   just `import caffe`.

 * You need to be running on one of the lab machines (ugrad1 through
   ugrad24; masters1 through masters16) in order to use GPU mode.  Caffe
   will run on our compute systems (ugradx, gradx, gradz), but only in CPU
   mode; you won't be able to do the GPU section of the notebook on those
   three systems.

 * The systemwide installation *only* includes the Caffe binaries and
   python modules.  It does not include the utility scripts and reference
   models.  You will need to download those yourself.

What you'll need to do is grab your own copy of the Caffe distribution in order
to use their models:

    git clone https://github.com/BVLC/caffe.git

Then, as the notebook indicates, run python from the examples directory:

    cd caffe/examples
    python

The notebook's second input block should be as follows:

    caffe_root = '../'  # this file should be run from {caffe_root}/examples (otherwise change this line)
    import caffe

Everything else should work as expected.